input: 'cg': nodes=species=Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, cart=Array([[-1.6074809 , 3.990498 , 0.59335387],
[13.654599 , 0.6031299 , 4.7416563 ],
[ 2.6521325 , 3.3873684 , 4.7416563 ],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]], dtype=float32):Rendering array...
, graph_i=Array([ 0, 0, 0, ..., 31, 31, 31], dtype=int32):Rendering array...
,), edges=to_jimage=Array([[[-1, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, -1]],
[[ 0, -1, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 1, 0, 0],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]],
...,
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]]], dtype=int8):Rendering array...
, receiver=Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
,), n_node=Array([168, 80, 68, 60, 53, 48, 42, 40, 36, 32, 31, 28, 26,
24, 24, 21, 20, 18, 16, 16, 14, 12, 12, 10, 9, 8,
6, 5, 4, 4, 1, 88], dtype=uint16):Rendering array...
, padding_mask=Array([ True, True, True, True, True, True, True, True, True,
True, True, True, True, True, True, True, True, True,
True, True, True, True, True, True, True, True, True,
True, True, True, True, False], dtype=bool):Rendering array...
, graph_data=dataset_id=Array([ 114536034, 1780206004, -1687275888, 1202003, 1758626004,
1556291002, 139460000, 1554815003, 118640006, 1675109005,
-1651451869, 1555003018, 1615850000, -1709938884, -1592934888,
-1650297867, -1696810885, -1663817880, -1696683861, -1710558886,
-1662911874, -1781222858, 123675020, -1806109888, -1804816886,
-1875814888, -1668181878, 122133000, 1864969002, -1700082886,
120483000, 0], dtype=int32):Rendering array...
, abc=Array([[17.914213 , 17.914213 , 8.296605 ],
[10.275203 , 16.84246 , 5.0821056],
[ 6.74365 , 9.197391 , 12.196357 ],
[ 9.433409 , 9.433409 , 20.947056 ],
[10.816606 , 10.426013 , 10.417979 ],
[ 6.8768296, 11.247085 , 19.015215 ],
[ 9.827564 , 9.827564 , 6.762959 ],
[ 6.386287 , 6.386287 , 12.2509365],
[ 8.614349 , 8.614349 , 8.614349 ],
[ 7.9092207, 7.7653685, 6.9275327],
[ 8.073368 , 20.393888 , 10.60539 ],
[ 5.306318 , 9.112615 , 10.040712 ],
[ 5.627906 , 6.953792 , 10.501416 ],
[ 5.037883 , 5.8846297, 7.8679 ],
[ 5.66 , 5.650074 , 9.156885 ],
[ 6.1716037, 8.621225 , 5.767166 ],
[ 6.2029986, 8.613454 , 8.613454 ],
[ 5.494526 , 5.494526 , 7.4353304],
[ 3.7546806, 4.811737 , 5.4126735],
[ 2.715433 , 5.2583437, 9.388806 ],
[ 6.0365143, 6.034208 , 6.02813 ],
[ 5.419802 , 6.49084 , 6.660221 ],
[ 4.117953 , 5.849566 , 5.849566 ],
[ 4.299559 , 4.299559 , 10.174184 ],
[ 4.235417 , 7.5434947, 7.5435066],
[ 6.565039 , 6.565039 , 6.565039 ],
[ 4.3139734, 5.5680494, 5.484443 ],
[ 5.9383936, 5.9383936, 5.9383936],
[ 4.4320164, 4.4320164, 4.4320164],
[ 3.640625 , 3.640625 , 6.0473022],
[ 3.571267 , 3.571267 , 3.571267 ],
[ 0. , 0. , 0. ]], dtype=float32):Rendering array...
, angles_rad=Array([[1.5707964, 1.5707964, 2.0943952],
[1.5707964, 1.5707964, 1.5707964],
[1.2343053, 1.5818529, 1.1629694],
[1.5707964, 1.5707964, 1.5707964],
[1.8930188, 1.8607063, 2.0151982],
[1.7593405, 1.5707964, 1.5707964],
[1.5707964, 1.5707964, 2.0943952],
[1.730846 , 1.730846 , 1.8141689],
[1.0478324, 1.0478324, 1.0478324],
[1.8424883, 1.5707964, 1.5707964],
[0.3288564, 1.0035257, 1.0851887],
[1.5707964, 1.5707964, 1.5707964],
[1.5707964, 1.5707964, 1.5707964],
[1.7750735, 1.7689804, 1.2668823],
[1.6448978, 1.4945464, 1.8871104],
[1.5798758, 1.5553586, 1.5304748],
[1.2806493, 1.2024467, 1.2024467],
[1.2456238, 1.397332 , 1.9114691],
[1.5707964, 1.5707964, 1.5707964],
[1.644386 , 1.5707964, 1.5707964],
[1.5746922, 2.094859 , 1.0429444],
[2.0565953, 1.3555512, 1.5138267],
[1.5707964, 1.5707964, 1.5707964],
[1.5707964, 1.5707964, 1.5707964],
[2.0943942, 1.5707964, 1.5707964],
[2.2965217, 2.2965217, 1.237292 ],
[1.6661328, 1.9750055, 1.1730014],
[2.4071865, 2.4071865, 1.0650531],
[1.0471976, 1.0471976, 1.0471976],
[1.5707964, 1.5707964, 2.0943952],
[1.0471976, 1.0471976, 1.0471976],
[0. , 0. , 0. ]], dtype=float32):Rendering array...
, lat=Array([[[ 1.7914213e+01, 1.5000001e-07, 0.0000000e+00],
[-8.9571066e+00, 1.5514163e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 8.2966051e+00]],
[[ 1.0275203e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.6842461e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 5.0821056e+00]],
[[ 1.6343699e-01, -1.7300400e-01, 6.7394490e+00],
[ 8.5276871e+00, 6.0145002e-02, 3.4448421e+00],
[ 4.2400112e+00, 1.1435485e+01, 5.5798002e-02]],
[[ 9.4334087e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 9.4334087e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 2.0947056e+01]],
[[-6.3495708e+00, 6.5125003e+00, 5.8539939e+00],
[ 6.0938725e+00, -6.4372239e+00, 5.4889545e+00],
[ 6.0836735e+00, 6.1772451e+00, -5.7762318e+00]],
[[ 6.8768296e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.1048177e+01, -2.1058772e+00],
[ 0.0000000e+00, -3.7062699e-03, 1.9015215e+01]],
[[ 4.9137821e+00, -8.5109196e+00, 0.0000000e+00],
[ 4.9137821e+00, 8.5109196e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 6.7629590e+00]],
[[ 6.3021402e+00, -6.3853199e-03, -1.0332731e+00],
[-1.7227671e+00, 6.0621023e+00, -1.0332731e+00],
[ 3.0169221e-02, 3.9895471e-02, 1.2250834e+01]],
[[ 7.4619780e+00, -1.4151999e-03, 4.3041716e+00],
[ 2.4855030e+00, 7.0358648e+00, 4.3041716e+00],
[-2.0012499e-03, -1.4151999e-03, 8.6143484e+00]],
[[ 7.9092207e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 7.5781007e+00, -1.6950918e+00],
[ 0.0000000e+00, -3.5751951e-01, 6.9183011e+00]],
[[ 6.7416244e+00, 5.3566109e-02, 4.4414983e+00],
[-1.2468886e+00, 7.0104933e+00, 1.9110441e+01],
[-1.6173536e-01, 2.7245510e-01, 1.0600656e+01]],
[[ 5.3063178e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 9.1126146e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0040712e+01]],
[[ 5.6279058e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 6.9537921e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0501416e+01]],
[[ 4.9857669e+00, 3.5377501e-03, -7.2275919e-01],
[ 1.6229881e+00, 5.5577154e+00, -1.0519388e+00],
[-4.2640892e-01, -7.8503363e-02, 7.8559442e+00]],
[[-4.6255998e-02, 4.7460151e+00, 3.0836339e+00],
[ 4.7340288e+00, -4.6193000e-02, -3.0838571e+00],
[-4.7798352e+00, 4.7923450e+00, -6.1672649e+00]],
[[ 6.1701546e+00, 1.2492373e-01, 4.7747571e-02],
[ 1.7339531e-01, 8.6193953e+00, -3.8590480e-02],
[ 4.4977799e-02, -2.7458400e-02, 5.7669253e+00]],
[[ 0.0000000e+00, 9.9999999e-09, -6.2029986e+00],
[-5.1461577e+00, 6.1716561e+00, -3.1015000e+00],
[ 5.1461577e+00, 6.1716561e+00, -3.1015000e+00]],
[[-3.9119451e+00, -3.1703908e+00, 2.1988914e+00],
[ 3.9119451e+00, -3.1703908e+00, -2.1988914e+00],
[ 3.8619483e+00, -3.1703908e+00, 5.5061889e+00]],
[[ 0.0000000e+00, -3.7546806e+00, 0.0000000e+00],
[-4.8117371e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -5.4126735e+00]],
[[ 2.7154329e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.2441120e+00, -3.8661101e-01],
[ 0.0000000e+00, 0.0000000e+00, 9.3888063e+00]],
[[-4.2616282e+00, -8.0763102e-03, -4.2752743e+00],
[-8.2218703e-03, -4.2585020e+00, -4.2751322e+00],
[-3.6448999e-03, -4.2542191e+00, 4.2708263e+00]],
[[ 5.3959742e+00, 2.2922444e-01, -4.5296320e-01],
[-7.0430361e-02, -1.8796436e+00, -6.2123251e+00],
[ 1.2653049e+00, 6.4092894e+00, 1.2955921e+00]],
[[ 4.1179528e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.8495660e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 5.8495660e+00]],
[[ 4.2995591e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 4.2995591e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0174184e+01]],
[[ 0.0000000e+00, 0.0000000e+00, -4.2354169e+00],
[-3.7717474e+00, -6.5328579e+00, 0.0000000e+00],
[-3.7717595e+00, 6.5328650e+00, 0.0000000e+00]],
[[-2.6921539e+00, 2.6921539e+00, 5.3483038e+00],
[ 2.6921539e+00, -2.6921539e+00, 5.3483038e+00],
[ 2.6921539e+00, 2.6921539e+00, -5.3483038e+00]],
[[-4.3139734e+00, 0.0000000e+00, 0.0000000e+00],
[-2.1569867e+00, -3.5078707e+00, 3.7477229e+00],
[ 2.1569867e+00, 3.4406567e+00, 3.6862457e+00]],
[[-2.1319211e+00, 2.1319211e+00, 5.1160870e+00],
[ 2.1319211e+00, -2.1319211e+00, 5.1160870e+00],
[ 2.1319211e+00, 2.1319211e+00, -5.1160870e+00]],
[[ 0.0000000e+00, 3.1339090e+00, 3.1339090e+00],
[ 3.1339090e+00, 0.0000000e+00, 3.1339090e+00],
[ 3.1339090e+00, 3.1339090e+00, 0.0000000e+00]],
[[ 1.8203124e+00, -3.1528738e+00, 0.0000000e+00],
[ 1.8203124e+00, 3.1528738e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 6.0473022e+00]],
[[ 0.0000000e+00, 2.5252669e+00, 2.5252669e+00],
[ 2.5252669e+00, 0.0000000e+00, 2.5252669e+00],
[ 2.5252669e+00, 2.5252669e+00, 0.0000000e+00]],
[[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00]]], dtype=float32):Rendering array...
,), target_data=e_form=Array([-6.45586 , -6.8128195, -6.207207 , -5.4671555, -6.7837076,
-5.1435127, -7.477968 , -8.923818 , -8.200472 , -7.9467635,
-9.577978 , -6.3009315, -6.341573 , -6.54503 , -9.78936 ,
-8.184659 , -6.8508086, -5.8719406, -4.560977 , -6.476559 ,
-8.410168 , -7.7028112, -4.6160893, -7.076347 , -4.2489 ,
-4.8315697, -4.6079087, -5.067987 , -4.573908 , -4.5372405,
-3.712647 , 0. ], dtype=float32):Rendering array...
, force=Array([[ 0.01931927, -0.00101004, 0.00514533],
[ 0.01053436, 0.01622596, 0.00514533],
[ 0.00878491, -0.017236 , 0.00514533],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]], dtype=float32):Rendering array...
, stress=Array([[[-5.7650656e-01, 1.3470000e-05, 0.0000000e+00],
[-1.3470000e-05, -5.7650650e-01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -1.8034226e-01]],
[[ 1.2198363e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.0798175e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 7.2570233e+00]],
[[-6.6261089e-01, -1.3693068e+00, 1.7417374e-01],
[-1.3692706e+00, 9.3778890e-01, 5.2438201e-03],
[ 1.7419136e-01, 5.2425899e-03, -9.8206145e-01]],
[[-2.8648193e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -2.8648193e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -1.8271584e+01]],
[[-3.5376236e+00, -6.9719963e-02, -1.5628552e-01],
[-6.9719352e-02, -3.4879370e+00, -6.0669428e-01],
[-1.5628555e-01, -6.0669398e-01, -2.6510825e+00]],
[[ 4.2228210e-01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 3.6939880e-01, 2.1828409e-01],
[ 0.0000000e+00, 2.1827926e-01, 7.3054400e-03]],
[[ 5.8439370e-02, 2.6000001e-07, -0.0000000e+00],
[-2.6000001e-07, 5.8439370e-02, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -1.5984979e+00]],
[[-5.6706039e+01, 2.8917530e+00, -1.5447336e+01],
[ 2.8917530e+00, -5.5068783e+01, -2.0427404e+01],
[-1.5447323e+01, -2.0427385e+01, -2.5514915e+01]],
[[-5.2726287e-01, -5.5939171e-02, -1.3698542e-01],
[-5.5939171e-02, -4.8771665e-01, -9.6870460e-02],
[-1.3698542e-01, -9.6870460e-02, -6.8537790e-01]],
[[ 5.0960288e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 2.7661177e+01, 8.4236950e-01],
[ 0.0000000e+00, 8.4236366e-01, -8.4243935e-01]],
[[-3.1354368e+00, -2.1732990e-02, -1.6962990e-02],
[-2.1732951e-02, -3.8033817e+00, 3.7452817e+00],
[-1.6962999e-02, 3.7452817e+00, -1.9895161e+00]],
[[ 1.7607585e-01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.0304039e-01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.6413042e-01]],
[[-2.0985847e+00, -1.6143000e-04, -4.7509999e-05],
[-1.6518999e-04, -1.3785554e+00, -1.3842000e-04],
[-5.1700001e-05, -1.3332001e-04, 8.3491117e-01]],
[[-9.3856964e+00, -3.4404862e-01, -2.0616198e+00],
[-3.4404868e-01, -7.5551205e+00, -7.3723370e-01],
[-2.0616195e+00, -7.3723364e-01, -1.1444527e+01]],
[[-1.9861711e+00, -2.8908322e+00, 7.8711361e-01],
[-2.8947139e+00, -4.8037186e+00, 2.2988582e-01],
[ 7.8879166e-01, 2.2843994e-01, 2.1315180e+01]],
[[-1.4272230e+00, -1.6913697e+00, -5.8565631e+00],
[-1.6913695e+00, -4.9773030e+00, -1.3547415e+00],
[-5.8565636e+00, -1.3547415e+00, 1.7153270e+00]],
[[-6.0113930e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -8.9604294e+01, 7.7699997e-06],
[ 0.0000000e+00, 7.7699997e-06, -6.6187057e+01]],
[[ 2.7452025e+00, -0.0000000e+00, 4.3098168e+00],
[-0.0000000e+00, 9.5339581e-02, 0.0000000e+00],
[ 4.3098168e+00, 0.0000000e+00, -4.1897740e+00]],
[[-1.9521265e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -6.4751107e-01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -7.5113559e-01]],
[[ 2.3531018e+02, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 2.1068205e+01, -1.0828968e+02],
[ 0.0000000e+00, -1.0828968e+02, 1.9093062e+02]],
[[-4.9796581e+00, -1.2185392e+00, -4.9542028e-01],
[-1.2196398e+00, -2.3808525e+00, -6.0947609e-01],
[-4.9716067e-01, -6.1348146e-01, -7.8811878e-01]],
[[ 8.0838509e+00, 1.0610949e+01, -6.8462425e-01],
[ 1.0610948e+01, 1.6280436e+01, -5.4885068e+00],
[-6.8462473e-01, -5.4885073e+00, -1.0486948e+01]],
[[ 1.4801001e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.6347537e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 5.6347537e+00]],
[[ 4.2995720e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 4.2995720e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 6.7877603e+00]],
[[-5.6527023e+00, -1.9787040e-02, 0.0000000e+00],
[-1.9787040e-02, -5.6755505e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -7.4912200e+00]],
[[-1.5253462e+00, 0.0000000e+00, -0.0000000e+00],
[ 0.0000000e+00, -1.5253462e+00, -0.0000000e+00],
[-0.0000000e+00, -0.0000000e+00, 3.4477546e+00]],
[[-1.2089447e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.6878867e+00, -2.0085368e+00],
[ 0.0000000e+00, -2.0085406e+00, 3.9861345e+00]],
[[ 1.3792354e-01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.3792354e-01, -0.0000000e+00],
[ 0.0000000e+00, -0.0000000e+00, 2.5862169e-01]],
[[ 1.4818479e+01, 1.4620157e+00, 1.4620157e+00],
[ 1.4620157e+00, 1.4818479e+01, 1.4620157e+00],
[ 1.4620157e+00, 1.4620157e+00, 1.4818479e+01]],
[[-3.0764341e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -3.0764344e+00, 0.0000000e+00],
[-0.0000000e+00, 0.0000000e+00, 4.8425870e+00]],
[[-1.4238837e+00, 0.0000000e+00, -0.0000000e+00],
[ 0.0000000e+00, -1.4238837e+00, 0.0000000e+00],
[-0.0000000e+00, 0.0000000e+00, -1.4238837e+00]],
[[ 1.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0000000e+00]]], dtype=float32):Rendering array...
,),),}children: 'mace': maceinput: 'vectors': Array([[[-0.08514988, -1.5187607 , 0.35024124],
[-1.322167 , 1.2215691 , 0.7999469 ],
[ 1.3537681 , -2.1077032 , 0.3074897 ],
...,
[ 3.0676713 , 3.6979523 , 0.7337971 ],
[ 1.6074809 , -3.990498 , 2.275238 ],
[ 0.06579673, 4.6275563 , -1.5481044 ]],
[[ 1.2727108 , -0.83312184, 0.35024118],
[-1.7189932 , -0.53424555, 0.7999468 ],
[ 2.5022097 , 0.11854595, 0.30748987],
...,
[-1.668685 , 4.505657 , 0.7337971 ],
[ 4.259614 , -0.60312974, 2.275238 ],
[-3.9746828 , 2.37076 , -1.548104 ]],
[[-1.3578603 , -0.68563867, 0.35024118],
[ 0.39682627, 1.7558143 , 0.7999468 ],
[-1.1484407 , -2.2262492 , 0.30748987],
...,
[ 4.736356 , -0.8077049 , 0.7337971 ],
[-2.6521325 , -3.3873684 , 2.275238 ],
[ 4.04048 , 2.2567966 , -1.548104 ]],
...,
[[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]],
[[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]],
[[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]]], dtype=float32):Rendering array...
, 'node_species': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
,}children: 'node_embedding': node_embeddinginput: 'node_species': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
,}children: 'embed': embedinput: 'inputs': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
,}params: 11520 parametersembedding: Array([[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-1.9298473e-01, 5.5534367e-02, 6.1264433e-02, ...,
7.9367173e-01, -6.8340577e-02, -2.3404005e-01],
[ 1.5960895e-01, -8.8038474e-02, 8.5871279e-02, ...,
9.9117720e-01, 1.6090073e-02, 1.2284823e-01],
...,
[ 4.0848142e-01, 4.4648898e-01, 1.2513540e-02, ...,
-3.9285350e-01, 8.4816426e-02, 1.3216788e-01],
[-7.4507594e-02, -1.0931188e+00, 8.0863041e-01, ...,
4.5552161e-01, 5.7509476e-01, 9.8894668e-01],
[-1.5040071e-01, -2.0703952e-01, 8.7037069e-01, ...,
-3.4091997e-01, -2.9804537e-01, -2.9061580e-01]], dtype=float32):Rendering array...
,}output: Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
,),}output: 128x0e
[[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
...
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]]key=0,): Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32), :Rendering array...
,,), 'edge_embedding': edge_embeddinginput: 'edge_lengths': Array([[1.5609463, 1.9698405, 2.5238166, ..., 4.860444 , 4.866701 ,
4.8800855],
[1.5609463, 1.9698402, 2.5238178, ..., 4.860444 , 4.8667016,
4.880085 ],
[1.5609465, 1.96984 , 2.5238168, ..., 4.8604436, 4.866701 ,
4.8800855],
...,
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ]], dtype=float32):Rendering array...
,}children: 'basis': basisinput: 'x': Array([[1.5609463, 1.9698405, 2.5238166, ..., 4.860444 , 4.866701 ,
4.8800855],
[1.5609463, 1.9698402, 2.5238178, ..., 4.860444 , 4.8667016,
4.880085 ],
[1.5609465, 1.96984 , 2.5238168, ..., 4.8604436, 4.866701 ,
4.8800855],
...,
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ]], dtype=float32):Rendering array...
, 'r_max': <jax.Array([5.], dtype=float32)>:Rendering array...
,}params: 8 parametersfreq: Array([0.37965846, 1.0474222 , 1.6848834 , 2.3603554 , 3.0224771 ,
3.7064195 , 4.424161 , 5.1440697 ], dtype=float32):Rendering array...
,}output: Array([[[ 0.14312896, 0.25572735, 0.12545101, ..., -0.12232928,
0.14945401, 0.25231254],
[ 0.13809775, 0.17893293, -0.03582633, ..., 0.17280407,
0.13234529, -0.13208821],
[ 0.12966888, 0.07571861, -0.14201179, ..., 0.01115712,
-0.1562002 , 0.06409264],
...,
[ 0.0792156 , -0.07647058, 0.07771403, ..., -0.06099052,
0.03856843, -0.01069126],
[ 0.07906061, -0.07617152, 0.07732467, ..., -0.05961608,
0.0364945 , -0.0080494 ],
[ 0.0787289 , -0.07552354, 0.07646646, ..., -0.05658155,
0.03198421, -0.00239656]],
[[ 0.14312896, 0.25572735, 0.12545101, ..., -0.12232928,
0.14945401, 0.25231254],
[ 0.13809775, 0.178933 , -0.03582628, ..., 0.17280397,
0.13234545, -0.13208795],
[ 0.12966886, 0.07571841, -0.14201187, ..., 0.01115651,
-0.15619996, 0.06409344],
...,
[ 0.0792156 , -0.07647058, 0.07771403, ..., -0.06099052,
0.03856843, -0.01069126],
[ 0.0790606 , -0.07617152, 0.07732463, ..., -0.05961608,
0.03649421, -0.00804925],
[ 0.0787289 , -0.07552354, 0.07646649, ..., -0.05658168,
0.0319845 , -0.00239672]],
[[ 0.14312896, 0.25572735, 0.12545094, ..., -0.12232928,
0.14945419, 0.25231248],
[ 0.13809773, 0.178933 , -0.03582619, ..., 0.17280397,
0.1323456 , -0.13208795],
[ 0.12966889, 0.07571857, -0.14201179, ..., 0.01115712,
-0.15620011, 0.0640929 ],
...,
[ 0.07921562, -0.0764706 , 0.07771403, ..., -0.06099075,
0.03856857, -0.01069141],
[ 0.07906061, -0.07617152, 0.07732467, ..., -0.05961608,
0.0364945 , -0.0080494 ],
[ 0.0787289 , -0.07552354, 0.07646646, ..., -0.05658155,
0.03198421, -0.00239656]],
...,
[[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
...,
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ]],
[[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
...,
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ]],
[[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
...,
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ],
[ 0.15186338, 0.4189689 , 0.67395335, ..., 1.4825678 ,
1.7696644 , 2.057628 ]]], dtype=float32):Rendering array...
,), 'envelope': envelopeinput: 'r': Array([[1.5609463, 1.9698405, 2.5238166, ..., 4.860444 , 4.866701 ,
4.8800855],
[1.5609463, 1.9698402, 2.5238178, ..., 4.860444 , 4.8667016,
4.880085 ],
[1.5609465, 1.96984 , 2.5238168, ..., 4.8604436, 4.866701 ,
4.8800855],
...,
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ],
[0. , 0. , 0. , ..., 0. , 0. ,
0. ]], dtype=float32):Rendering array...
, 'r_max': <jax.Array([5.], dtype=float32)>:Rendering array...
,}output: Array([[1. , 1. , 1. , ..., 0.20315367, 0.18754505,
0.15560311],
[1. , 1. , 1. , ..., 0.20315367, 0.18754408,
0.1556041 ],
[1. , 1. , 1. , ..., 0.20315467, 0.18754505,
0.15560311],
...,
[1. , 1. , 1. , ..., 1. , 1. ,
1. ],
[1. , 1. , 1. , ..., 1. , 1. ,
1. ],
[1. , 1. , 1. , ..., 1. , 1. ,
1. ]], dtype=float32):Rendering array...
,),}output: 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,,), 'edge_embedding.avg_num_neighbors': edge_embeddinginput: 'metadata': dataset_name='mptrj', supported_targets='0': 'energy',}, batches_per_group=array([ 906, 1425, 1409, 1425, 1438, 1440, 1436, 1440, 1422, 1430, 1445,
1492, 1398, 1439, 1425, 1443, 1465, 1413, 1397, 1394, 1425, 1475,
1492, 1430, 1388, 1445, 1411, 1460, 1491, 1433, 1451, 1420, 1464,
1427, 1402, 1430], dtype=uint32):Rendering array...
, batch_num_atoms=32, nearest_k=48, batch_num_graphs=32, shift_energy=-6.198672771453857, scale_energy=2.3766870498657227, atomic_numbers=array([ 0, 6, 17, 3, 8, 15, 51, 12, 26, 29, 21, 31, 37, 83, 25, 23, 24,
16, 62, 32, 46, 34, 70, 19, 60, 9, 28, 7, 14, 20, 30, 52, 42, 27,
56, 38, 72, 41, 39, 47, 48, 50, 90, 40, 92, 58, 45, 55, 13, 53, 33,
82, 5, 22, 49, 11, 35, 78, 69, 73, 57, 93, 1, 94, 66, 77, 81, 4,
74, 64, 63, 75, 43, 59, 79, 44, 68, 80, 65, 67, 76, 36, 91, 71, 61,
54, 89, 10, 2, 18], dtype=uint8):Rendering array...
, atomwise_shift_energy=array([ 0. , -2.2707615 , 2.3401892 , 3.0822957 ,
-2.139031 , -0.16549587, 2.1411645 , 4.8762283 ,
-1.9809715 , 2.5143476 , -1.9116923 , 2.9952776 ,
4.3576717 , 2.7872267 , -2.9721358 , -3.6167579 ,
-3.5604646 , 0.6107417 , -1.0343803 , 1.5463392 ,
0.69618833, 1.4034783 , 2.7666323 , 4.2159247 ,
-0.9729802 , 0.4112059 , 0.38132888, -2.0058405 ,
0.34989974, 2.1912303 , 5.0034466 , 2.3049965 ,
-4.728218 , -0.54687583, 1.999384 , 2.2432775 ,
-5.1602435 , -4.790666 , -2.4661624 , 4.0280995 ,
5.34998 , 2.2807906 , -3.1929913 , -3.7777352 ,
-6.3766856 , -1.6896563 , -1.404504 , 4.1458993 ,
2.040268 , 3.405083 , 1.543168 , 2.8831744 ,
-1.0880722 , -3.9293804 , 3.4315777 , 4.092721 ,
2.8371766 , -0.34215775, -0.4136282 , -6.77352 ,
-1.7962555 , -6.870695 , 2.6937704 , -8.061367 ,
-0.5946495 , -2.793373 , 4.1255984 , 2.1051993 ,
-6.384699 , -10.322368 , -6.0671163 , -5.471272 ,
-1.4151858 , -0.7534348 , 2.9328883 , -2.8679254 ,
-0.39531928, 6.6045566 , -0.5138772 , -0.4270394 ,
-4.0246854 , 6.1608987 , -4.3657956 , -0.40262896,
0.736037 , 16.102348 , 1.9727042 , 0.7799125 ,
3.719906 , 1.1950303 ], dtype=float32):Rendering array...
, atomwise_scale_energy=array([-2.40991106e-39, 1.33058935e-01, 2.93497473e-01, 2.61184499e-02,
2.40968168e-01, 9.34929773e-02, 3.21534164e-02, 1.22204259e-01,
-3.99079267e-03, 3.21241319e-02, 5.05398333e-01, 1.65204424e-02,
1.08429275e-01, 1.10232485e-02, -2.18620170e-02, -2.22697780e-02,
-2.75362208e-02, 1.05140291e-01, 2.78376997e-01, 2.12133881e-02,
6.50004968e-02, 7.48609379e-02, 1.90245599e-01, 1.06534272e-01,
3.27279478e-01, 2.27959231e-01, 2.23455299e-02, 4.60880756e-01,
1.62185147e-01, 1.83875978e-01, 2.92971972e-02, 4.69289832e-02,
1.18544335e-02, 3.74471620e-02, 1.70294091e-01, 1.80884063e-01,
4.55600560e-01, 1.10464349e-01, 3.83205593e-01, 8.41733068e-02,
-6.63747604e-04, 1.12307980e-03, 3.71447802e-01, 3.02267253e-01,
1.66967139e-01, 2.51082033e-01, 1.04348332e-01, 8.47445428e-02,
1.59817740e-01, 1.23538606e-01, 7.84063041e-02, 8.03675130e-03,
1.46966413e-01, 2.89947659e-01, -1.36596169e-02, 7.84631744e-02,
1.75462380e-01, 5.37232794e-02, 2.80312657e-01, 2.79683918e-01,
3.19721878e-01, 2.42589667e-01, -3.06586875e-03, 2.37827122e-01,
2.86706179e-01, 7.04936832e-02, 3.83133627e-02, 6.03754342e-01,
-9.21594165e-03, 2.57200658e-01, 1.97187737e-01, 3.15266103e-01,
1.49181616e+00, 3.04980606e-01, 4.43397984e-02, 5.12347743e-02,
2.87493646e-01, 1.02658063e-01, 2.45865300e-01, 2.92902708e-01,
2.93616891e-01, 1.51683092e+00, 3.58057797e-01, 3.03377599e-01,
3.02371353e-01, 1.40909803e+00, 2.45032728e-01, 1.02921605e+00,
-7.56268263e-01, 1.25650644e+00], dtype=float32):Rendering array...
, r_max_quantile_r=array([ 0.5, 1. , 1.5, 2. , 2.5, 3. , 3.5, 4. , 4.5, 5. , 5.5,
6. , 6.5, 7. , 7.5, 8. , 8.5, 9. , 9.5, 10. , 10.5, 11. ,
11.5, 12. ], dtype=float32):Rendering array...
, r_max_quantile_k=array([[9.9999923e-01, 7.3028741e-07, 0.0000000e+00, ..., 0.0000000e+00,
0.0000000e+00, 0.0000000e+00],
[9.6614134e-01, 2.8218286e-02, 5.4964880e-03, ..., 0.0000000e+00,
0.0000000e+00, 0.0000000e+00],
[8.0479968e-01, 1.3872136e-01, 2.9714543e-02, ..., 0.0000000e+00,
0.0000000e+00, 0.0000000e+00],
...,
[1.4321747e-05, 4.8482966e-06, 1.3611745e-05, ..., 5.5298984e-05,
9.6763079e-06, 9.9777752e-01],
[9.0880203e-06, 3.8137230e-06, 1.0284881e-05, ..., 9.9745084e-05,
1.4930320e-05, 9.9792618e-01],
[6.5523009e-06, 2.3937198e-06, 7.8100175e-06, ..., 1.5727550e-04,
2.5276058e-05, 9.9818617e-01]], dtype=float32):Rendering array...
,),}output: <jax.Array([35.006073], dtype=float32)>:Rendering array...
,), 'layer_0': layer_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e
[[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
...
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]]key=0,): Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32), :Rendering array...
,, 'node_species': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'interaction': interactioninput: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e
[[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
...
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]]key=0,): Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32), :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'resid_adapter': resid_adapterinput: 'weights_or_input': 128x0e
[[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
...
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]]key=0,): Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32), :Rendering array...
,,}params: 28672 parametersw[0,0] 128x0e,224x0e: Array([[-0.15340874, 0.63586867, -2.066119 , ..., -1.2421888 ,
-0.564434 , -1.1749403 ],
[-0.4223909 , 1.9502263 , -1.2073642 , ..., 0.26299307,
1.3463928 , -0.4028095 ],
[ 1.0757334 , 0.45307207, 0.2272922 , ..., 1.7615929 ,
-1.8404636 , 0.5965187 ],
...,
[-1.5144392 , 1.1170446 , -1.6822397 , ..., -1.7941395 ,
-1.1314518 , -0.31909847],
[-0.03311044, 0.73373675, -0.40699175, ..., 0.63367414,
0.6495996 , -0.9303382 ],
[ 0.27654043, 0.6394971 , -0.8532704 , ..., -0.43343392,
-1.7429564 , 2.0256884 ]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[ 1.7519009e-01 7.2698981e-02 1.2544623e-01 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.7519009e-01 7.2698981e-02 1.2544623e-01 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.7519009e-01 7.2698981e-02 1.2544623e-01 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
...
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 1.7519009e-01, 7.2698981e-02, 1.2544623e-01, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.7519009e-01, 7.2698981e-02, 1.2544623e-01, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.7519009e-01, 7.2698981e-02, 1.2544623e-01, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
...,
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
,,), 'SimpleInteraction_0': SimpleInteraction_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e
[[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
...
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]]key=0,): Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32), :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'linear_intro': linear_introinput: 'weights_or_input': 128x0e
[[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
[ 4.2173825e-03 -7.3911045e-03 -1.3092387e-02 ... 6.2542468e-01
1.6457355e-02 -3.5454072e-02]
...
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]
[-4.2038954e-45 -4.2038954e-45 -4.2038954e-45 ... -4.2038954e-45
4.2038954e-45 -4.2038954e-45]]key=0,): Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32), :Rendering array...
,,}params: 16384 parametersw[0,0] 128x0e,128x0e: Array([[-0.61885774, -1.0856944 , -1.6025343 , ..., -0.55894697,
-1.3429723 , 0.34905353],
[ 0.09641334, 0.12247575, 1.810319 , ..., 0.14283091,
1.1057113 , 0.82308096],
[-1.4454341 , -2.1964498 , 0.57032555, ..., 0.59593636,
2.0616703 , 1.0615976 ],
...,
[ 1.7831123 , -0.78575855, 0.907529 , ..., -2.4380598 ,
0.76305825, -1.559743 ],
[ 0.5572579 , -0.3384806 , -0.7951938 , ..., -0.7975023 ,
1.2298973 , -1.0195863 ],
[-0.37849867, -0.37187925, 2.0620265 , ..., 4.0443935 ,
1.5165552 , -0.24680647]], dtype=float32):Rendering array...
,}output: 128x0e
[[-6.3483499e-02 3.1603071e-01 3.5126447e-03 ... -3.8959205e-01
1.9541387e-02 -3.8026479e-01]
[-6.3483499e-02 3.1603071e-01 3.5126447e-03 ... -3.8959205e-01
1.9541387e-02 -3.8026479e-01]
[-6.3483499e-02 3.1603071e-01 3.5126447e-03 ... -3.8959205e-01
1.9541387e-02 -3.8026479e-01]
...
[-5.6051939e-45 4.2038954e-45 0.0000000e+00 ... 0.0000000e+00
-2.8025969e-45 1.4012985e-45]
[-5.6051939e-45 4.2038954e-45 0.0000000e+00 ... 0.0000000e+00
-2.8025969e-45 1.4012985e-45]
[-5.6051939e-45 4.2038954e-45 0.0000000e+00 ... 0.0000000e+00
-2.8025969e-45 1.4012985e-45]]key=0,): Array([[-6.3483499e-02, 3.1603071e-01, 3.5126447e-03, ...,
-3.8959205e-01, 1.9541387e-02, -3.8026479e-01],
[-6.3483499e-02, 3.1603071e-01, 3.5126447e-03, ...,
-3.8959205e-01, 1.9541387e-02, -3.8026479e-01],
[-6.3483499e-02, 3.1603071e-01, 3.5126447e-03, ...,
-3.8959205e-01, 1.9541387e-02, -3.8026479e-01],
...,
[-5.6051939e-45, 4.2038954e-45, 0.0000000e+00, ...,
0.0000000e+00, -2.8025969e-45, 1.4012985e-45],
[-5.6051939e-45, 4.2038954e-45, 0.0000000e+00, ...,
0.0000000e+00, -2.8025969e-45, 1.4012985e-45],
[-5.6051939e-45, 4.2038954e-45, 0.0000000e+00, ...,
0.0000000e+00, -2.8025969e-45, 1.4012985e-45]], dtype=float32), :Rendering array...
,,), 'SevenNetConv_0': SevenNetConv_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e
[[-6.3483499e-02 3.1603071e-01 3.5126447e-03 ... -3.8959205e-01
1.9541387e-02 -3.8026479e-01]
[-6.3483499e-02 3.1603071e-01 3.5126447e-03 ... -3.8959205e-01
1.9541387e-02 -3.8026479e-01]
[-6.3483499e-02 3.1603071e-01 3.5126447e-03 ... -3.8959205e-01
1.9541387e-02 -3.8026479e-01]
...
[-5.6051939e-45 4.2038954e-45 0.0000000e+00 ... 0.0000000e+00
-2.8025969e-45 1.4012985e-45]
[-5.6051939e-45 4.2038954e-45 0.0000000e+00 ... 0.0000000e+00
-2.8025969e-45 1.4012985e-45]
[-5.6051939e-45 4.2038954e-45 0.0000000e+00 ... 0.0000000e+00
-2.8025969e-45 1.4012985e-45]]key=0,): Array([[-6.3483499e-02, 3.1603071e-01, 3.5126447e-03, ...,
-3.8959205e-01, 1.9541387e-02, -3.8026479e-01],
[-6.3483499e-02, 3.1603071e-01, 3.5126447e-03, ...,
-3.8959205e-01, 1.9541387e-02, -3.8026479e-01],
[-6.3483499e-02, 3.1603071e-01, 3.5126447e-03, ...,
-3.8959205e-01, 1.9541387e-02, -3.8026479e-01],
...,
[-5.6051939e-45, 4.2038954e-45, 0.0000000e+00, ...,
0.0000000e+00, -2.8025969e-45, 1.4012985e-45],
[-5.6051939e-45, 4.2038954e-45, 0.0000000e+00, ...,
0.0000000e+00, -2.8025969e-45, 1.4012985e-45],
[-5.6051939e-45, 4.2038954e-45, 0.0000000e+00, ...,
0.0000000e+00, -2.8025969e-45, 1.4012985e-45]], dtype=float32), :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'LazyInMLP_0': LazyInMLP_0input: 'x': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,,}children: 'Dense_0': Dense_0input: 'inputs': Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32):Rendering array...
,}params: 3072 parameterskernel: Array([[-1.6549976 , 1.8082368 , -0.931092 , ..., -1.1570392 ,
0.29403457, 2.0080574 ],
[-3.9210534 , 4.3420377 , -1.6388535 , ..., 0.42035243,
1.5823126 , 1.8510852 ],
[-5.891932 , 1.6466467 , -0.91515696, ..., 3.1036937 ,
0.06468228, -0.01008435],
...,
[-0.3021125 , 0.21281758, 2.282933 , ..., 2.4923284 ,
2.219395 , 0.38834897],
[ 1.2368158 , -0.5212946 , 1.5847732 , ..., 1.4557836 ,
0.837641 , 0.28724718],
[ 0.61822885, -0.6755894 , 0.49171826, ..., 0.6699977 ,
-0.3290476 , 0.2587202 ]], dtype=float32):Rendering array...
,}output: Array([[[-7.04744011e-02, 1.21852076e+00, -9.89617705e-01, ...,
-9.95731533e-01, -7.08278120e-01, 3.97574782e-01],
[ 6.35177255e-01, 1.03016412e+00, -1.49667375e-02, ...,
-8.23826373e-01, 3.75961483e-01, 4.20294553e-01],
[-3.30404937e-02, 4.40512866e-01, -1.09523706e-01, ...,
-4.86856103e-01, 3.95714074e-01, 5.15792191e-01],
...,
[-8.95373151e-03, -9.49798059e-03, -6.59004319e-03, ...,
-2.43883822e-02, -1.93166938e-02, 3.83972726e-03],
[-8.40501953e-03, -8.86404049e-03, -6.06027059e-03, ...,
-2.22131237e-02, -1.79946870e-02, 3.61366617e-03],
[-7.22509157e-03, -7.52223935e-03, -4.98296646e-03, ...,
-1.78870130e-02, -1.52160246e-02, 3.11819348e-03]],
[[-7.04744011e-02, 1.21852076e+00, -9.89617705e-01, ...,
-9.95731533e-01, -7.08278120e-01, 3.97574782e-01],
[ 6.35177970e-01, 1.03016400e+00, -1.49669414e-02, ...,
-8.23826849e-01, 3.75960767e-01, 4.20294493e-01],
[-3.30404937e-02, 4.40511018e-01, -1.09523691e-01, ...,
-4.86855090e-01, 3.95713001e-01, 5.15792251e-01],
...,
[-8.95373151e-03, -9.49798059e-03, -6.59004319e-03, ...,
-2.43883822e-02, -1.93166938e-02, 3.83972726e-03],
[-8.40497576e-03, -8.86400975e-03, -6.06032414e-03, ...,
-2.22131051e-02, -1.79946907e-02, 3.61362263e-03],
[-7.22511625e-03, -7.52228592e-03, -4.98296879e-03, ...,
-1.78871062e-02, -1.52160879e-02, 3.11822048e-03]],
[[-7.04736188e-02, 1.21852052e+00, -9.89617407e-01, ...,
-9.95731711e-01, -7.08278000e-01, 3.97574872e-01],
[ 6.35177970e-01, 1.03016400e+00, -1.49669452e-02, ...,
-8.23826671e-01, 3.75960618e-01, 4.20294344e-01],
[-3.30405124e-02, 4.40512449e-01, -1.09523349e-01, ...,
-4.86855447e-01, 3.95714134e-01, 5.15792370e-01],
...,
[-8.95374455e-03, -9.49803274e-03, -6.59012236e-03, ...,
-2.43885834e-02, -1.93168186e-02, 3.83974402e-03],
[-8.40501953e-03, -8.86404049e-03, -6.06027059e-03, ...,
-2.22131237e-02, -1.79946870e-02, 3.61366617e-03],
[-7.22509157e-03, -7.52223935e-03, -4.98296646e-03, ...,
-1.78870130e-02, -1.52160246e-02, 3.11819348e-03]],
...,
[[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
...,
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00]],
[[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
...,
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00]],
[[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
...,
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00]]], dtype=float32):Rendering array...
,),}output: 384x0e
[[[-7.04744011e-02 1.21852076e+00 -9.89617705e-01 ... -9.95731533e-01
-7.08278120e-01 3.97574782e-01]
[ 6.35177255e-01 1.03016412e+00 -1.49667375e-02 ... -8.23826373e-01
3.75961483e-01 4.20294553e-01]
[-3.30404937e-02 4.40512866e-01 -1.09523706e-01 ... -4.86856103e-01
3.95714074e-01 5.15792191e-01]
...
[-8.95373151e-03 -9.49798059e-03 -6.59004319e-03 ... -2.43883822e-02
-1.93166938e-02 3.83972726e-03]
[-8.40501953e-03 -8.86404049e-03 -6.06027059e-03 ... -2.22131237e-02
-1.79946870e-02 3.61366617e-03]
[-7.22509157e-03 -7.52223935e-03 -4.98296646e-03 ... -1.78870130e-02
-1.52160246e-02 3.11819348e-03]]
[[-7.04744011e-02 1.21852076e+00 -9.89617705e-01 ... -9.95731533e-01
-7.08278120e-01 3.97574782e-01]
[ 6.35177970e-01 1.03016400e+00 -1.49669414e-02 ... -8.23826849e-01
3.75960767e-01 4.20294493e-01]
[-3.30404937e-02 4.40511018e-01 -1.09523691e-01 ... -4.86855090e-01
3.95713001e-01 5.15792251e-01]
...
[-8.95373151e-03 -9.49798059e-03 -6.59004319e-03 ... -2.43883822e-02
-1.93166938e-02 3.83972726e-03]
[-8.40497576e-03 -8.86400975e-03 -6.06032414e-03 ... -2.22131051e-02
-1.79946907e-02 3.61362263e-03]
[-7.22511625e-03 -7.52228592e-03 -4.98296879e-03 ... -1.78871062e-02
-1.52160879e-02 3.11822048e-03]]
[[-7.04736188e-02 1.21852052e+00 -9.89617407e-01 ... -9.95731711e-01
-7.08278000e-01 3.97574872e-01]
[ 6.35177970e-01 1.03016400e+00 -1.49669452e-02 ... -8.23826671e-01
3.75960618e-01 4.20294344e-01]
[-3.30405124e-02 4.40512449e-01 -1.09523349e-01 ... -4.86855447e-01
3.95714134e-01 5.15792370e-01]
...
[-8.95374455e-03 -9.49803274e-03 -6.59012236e-03 ... -2.43885834e-02
-1.93168186e-02 3.83974402e-03]
[-8.40501953e-03 -8.86404049e-03 -6.06027059e-03 ... -2.22131237e-02
-1.79946870e-02 3.61366617e-03]
[-7.22509157e-03 -7.52223935e-03 -4.98296646e-03 ... -1.78870130e-02
-1.52160246e-02 3.11819348e-03]]
...
[[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
...
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]]
[[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
...
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]]
[[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
...
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]
[-1.18387394e+01 1.54664505e+00 7.77936172e+00 ... 1.79073753e+01
9.71461201e+00 4.97063732e+00]]]key=0,): Array([[[-7.04744011e-02, 1.21852076e+00, -9.89617705e-01, ...,
-9.95731533e-01, -7.08278120e-01, 3.97574782e-01],
[ 6.35177255e-01, 1.03016412e+00, -1.49667375e-02, ...,
-8.23826373e-01, 3.75961483e-01, 4.20294553e-01],
[-3.30404937e-02, 4.40512866e-01, -1.09523706e-01, ...,
-4.86856103e-01, 3.95714074e-01, 5.15792191e-01],
...,
[-8.95373151e-03, -9.49798059e-03, -6.59004319e-03, ...,
-2.43883822e-02, -1.93166938e-02, 3.83972726e-03],
[-8.40501953e-03, -8.86404049e-03, -6.06027059e-03, ...,
-2.22131237e-02, -1.79946870e-02, 3.61366617e-03],
[-7.22509157e-03, -7.52223935e-03, -4.98296646e-03, ...,
-1.78870130e-02, -1.52160246e-02, 3.11819348e-03]],
[[-7.04744011e-02, 1.21852076e+00, -9.89617705e-01, ...,
-9.95731533e-01, -7.08278120e-01, 3.97574782e-01],
[ 6.35177970e-01, 1.03016400e+00, -1.49669414e-02, ...,
-8.23826849e-01, 3.75960767e-01, 4.20294493e-01],
[-3.30404937e-02, 4.40511018e-01, -1.09523691e-01, ...,
-4.86855090e-01, 3.95713001e-01, 5.15792251e-01],
...,
[-8.95373151e-03, -9.49798059e-03, -6.59004319e-03, ...,
-2.43883822e-02, -1.93166938e-02, 3.83972726e-03],
[-8.40497576e-03, -8.86400975e-03, -6.06032414e-03, ...,
-2.22131051e-02, -1.79946907e-02, 3.61362263e-03],
[-7.22511625e-03, -7.52228592e-03, -4.98296879e-03, ...,
-1.78871062e-02, -1.52160879e-02, 3.11822048e-03]],
[[-7.04736188e-02, 1.21852052e+00, -9.89617407e-01, ...,
-9.95731711e-01, -7.08278000e-01, 3.97574872e-01],
[ 6.35177970e-01, 1.03016400e+00, -1.49669452e-02, ...,
-8.23826671e-01, 3.75960618e-01, 4.20294344e-01],
[-3.30405124e-02, 4.40512449e-01, -1.09523349e-01, ...,
-4.86855447e-01, 3.95714134e-01, 5.15792370e-01],
...,
[-8.95374455e-03, -9.49803274e-03, -6.59012236e-03, ...,
-2.43885834e-02, -1.93168186e-02, 3.83974402e-03],
[-8.40501953e-03, -8.86404049e-03, -6.06027059e-03, ...,
-2.22131237e-02, -1.79946870e-02, 3.61366617e-03],
[-7.22509157e-03, -7.52223935e-03, -4.98296646e-03, ...,
-1.78870130e-02, -1.52160246e-02, 3.11819348e-03]],
...,
[[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
...,
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00]],
[[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
...,
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00]],
[[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
...,
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00],
[-1.18387394e+01, 1.54664505e+00, 7.77936172e+00, ...,
1.79073753e+01, 9.71461201e+00, 4.97063732e+00]]], dtype=float32), :Rendering array...
,,),}output: 128x0e+128x1e+128x2e
[[-0.00085128 -0.043733 0.04078883 ... -0.13236058 0.01714112
-0.00657166]
[-0.00085128 -0.043733 0.04078883 ... 0.02513262 0.08785425
-0.09750038]
[-0.00085128 -0.043733 0.04078883 ... -0.00057919 -0.10499544
-0.08265567]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.00085128, -0.043733 , 0.04078883, ..., -0.13236058,
0.01714112, -0.00657166],
[-0.00085128, -0.043733 , 0.04078883, ..., 0.02513262,
0.08785425, -0.09750038],
[-0.00085128, -0.043733 , 0.04078883, ..., -0.00057919,
-0.10499544, -0.08265567],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'linear_outro': linear_outroinput: 'weights_or_input': 128x0e+128x1e+128x2e
[[-0.00085128 -0.043733 0.04078883 ... -0.13236058 0.01714112
-0.00657166]
[-0.00085128 -0.043733 0.04078883 ... 0.02513262 0.08785425
-0.09750038]
[-0.00085128 -0.043733 0.04078883 ... -0.00057919 -0.10499544
-0.08265567]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.00085128, -0.043733 , 0.04078883, ..., -0.13236058,
0.01714112, -0.00657166],
[-0.00085128, -0.043733 , 0.04078883, ..., 0.02513262,
0.08785425, -0.09750038],
[-0.00085128, -0.043733 , 0.04078883, ..., -0.00057919,
-0.10499544, -0.08265567],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 40960 parametersw[0,0] 128x0e,224x0e: Array([[ 8.8953382e-01, -9.6430862e-01, 1.4592638e+00, ...,
-6.1619239e+00, 3.9559295e+00, 1.8397622e+00],
[-5.0314933e-01, -1.6821375e+00, 2.0433755e+00, ...,
2.2306583e+00, 8.4645426e-01, -3.7939497e-04],
[-1.2256092e+00, 1.8113122e+00, -4.3403234e+00, ...,
-4.2426414e+00, -6.9003034e-01, 1.6837559e+00],
...,
[ 8.4209189e-02, -2.2738113e+00, -1.4586139e+00, ...,
-2.5382233e-01, -1.3814783e+00, 2.9294884e+00],
[-1.9275451e+00, -3.3198323e+00, -1.3414438e+00, ...,
5.2309602e-02, 3.8918704e-01, -4.7204347e+00],
[-9.1071028e-01, -1.0141186e+00, 9.5095253e-01, ...,
2.3143175e+00, 4.6380571e-01, -2.0695004e+00]], dtype=float32):Rendering array...
, w[1,1] 128x1e,64x1e: Array([[ 2.08344 , 2.0268428 , -0.10205896, ..., -0.6795716 ,
-0.9759364 , -2.71261 ],
[-1.3094982 , -1.9427985 , 1.2594589 , ..., -1.572477 ,
1.3153887 , 1.0310308 ],
[-0.09860689, 1.2901495 , 0.44501492, ..., 1.7403201 ,
-3.5873814 , 0.12359422],
...,
[ 2.3171134 , -0.4278286 , -2.723649 , ..., 2.31806 ,
4.9230413 , -3.0693502 ],
[ 1.5557939 , -2.9966815 , -2.6616237 , ..., -0.07412253,
-1.840733 , -0.17063624],
[ 1.3175749 , 0.1819717 , -0.4712196 , ..., -0.40548873,
1.1541386 , -1.3347238 ]], dtype=float32):Rendering array...
, w[2,2] 128x2e,32x2e: Array([[ 2.1402237 , -1.4459054 , -1.2280812 , ..., 0.5808771 ,
4.719111 , 0.23272462],
[ 0.5723715 , -2.2097077 , 0.5408782 , ..., 3.674021 ,
2.320326 , -0.91067314],
[ 1.716137 , 0.7318912 , 5.4104414 , ..., -0.8077885 ,
-0.01075178, -2.623032 ],
...,
[ 0.92815876, 1.0031501 , 5.834125 , ..., 1.7627509 ,
1.8706713 , -0.13664617],
[ 1.9301639 , -2.993204 , -5.7063103 , ..., 0.8726918 ,
0.07637297, -0.6783958 ],
[ 1.5137278 , -0.7742239 , -2.7060916 , ..., -1.565252 ,
3.3198183 , -1.4513925 ]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[-3.4928432e-01 -1.6145222e-01 -4.7439441e-01 ... 1.1000536e-01
2.7664907e-02 -1.0699533e-04]
[-3.4928697e-01 -1.6145234e-01 -4.7439048e-01 ... 6.9282344e-03
-9.6383870e-02 5.9404194e-02]
[-3.4928453e-01 -1.6145249e-01 -4.7439346e-01 ... -3.4569088e-02
6.8718247e-02 8.3363116e-02]
...
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-3.4928432e-01, -1.6145222e-01, -4.7439441e-01, ...,
1.1000536e-01, 2.7664907e-02, -1.0699533e-04],
[-3.4928697e-01, -1.6145234e-01, -4.7439048e-01, ...,
6.9282344e-03, -9.6383870e-02, 5.9404194e-02],
[-3.4928453e-01, -1.6145249e-01, -4.7439346e-01, ...,
-3.4569088e-02, 6.8718247e-02, 8.3363116e-02],
...,
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[-3.4928432e-01 -1.6145222e-01 -4.7439441e-01 ... 1.1000536e-01
2.7664907e-02 -1.0699533e-04]
[-3.4928697e-01 -1.6145234e-01 -4.7439048e-01 ... 6.9282344e-03
-9.6383870e-02 5.9404194e-02]
[-3.4928453e-01 -1.6145249e-01 -4.7439346e-01 ... -3.4569088e-02
6.8718247e-02 8.3363116e-02]
...
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 0.0000000e+00 0.0000000e+00 0.0000000e+00 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-3.4928432e-01, -1.6145222e-01, -4.7439441e-01, ...,
1.1000536e-01, 2.7664907e-02, -1.0699533e-04],
[-3.4928697e-01, -1.6145234e-01, -4.7439048e-01, ...,
6.9282344e-03, -9.6383870e-02, 5.9404194e-02],
[-3.4928453e-01, -1.6145249e-01, -4.7439346e-01, ...,
-3.4569088e-02, 6.8718247e-02, 8.3363116e-02],
...,
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[-1.7409423e-01 -8.8753238e-02 -3.4894818e-01 ... 1.1000536e-01
2.7664907e-02 -1.0699533e-04]
[-1.7409688e-01 -8.8753358e-02 -3.4894425e-01 ... 6.9282344e-03
-9.6383870e-02 5.9404194e-02]
[-1.7409444e-01 -8.8753507e-02 -3.4894723e-01 ... -3.4569088e-02
6.8718247e-02 8.3363116e-02]
...
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-1.7409423e-01, -8.8753238e-02, -3.4894818e-01, ...,
1.1000536e-01, 2.7664907e-02, -1.0699533e-04],
[-1.7409688e-01, -8.8753358e-02, -3.4894425e-01, ...,
6.9282344e-03, -9.6383870e-02, 5.9404194e-02],
[-1.7409444e-01, -8.8753507e-02, -3.4894723e-01, ...,
-3.4569088e-02, 6.8718247e-02, 8.3363116e-02],
...,
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'self_connection': self_connectioninput: 'node_feats': 224x0e+64x1e+32x2e
[[-1.7409423e-01 -8.8753238e-02 -3.4894818e-01 ... 1.1000536e-01
2.7664907e-02 -1.0699533e-04]
[-1.7409688e-01 -8.8753358e-02 -3.4894425e-01 ... 6.9282344e-03
-9.6383870e-02 5.9404194e-02]
[-1.7409444e-01 -8.8753507e-02 -3.4894723e-01 ... -3.4569088e-02
6.8718247e-02 8.3363116e-02]
...
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-1.7409423e-01, -8.8753238e-02, -3.4894818e-01, ...,
1.1000536e-01, 2.7664907e-02, -1.0699533e-04],
[-1.7409688e-01, -8.8753358e-02, -3.4894425e-01, ...,
6.9282344e-03, -9.6383870e-02, 5.9404194e-02],
[-1.7409444e-01, -8.8753507e-02, -3.4894723e-01, ...,
-3.4569088e-02, 6.8718247e-02, 8.3363116e-02],
...,
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_specie': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[-1.3326846e-01 -7.1100801e-02 -2.4199210e-01 ... 1.1515099e-01
2.8958965e-02 -1.1200016e-04]
[-1.3327029e-01 -7.1100898e-02 -2.4198994e-01 ... 7.2522336e-03
-1.0089127e-01 6.2182236e-02]
[-1.3326859e-01 -7.1101010e-02 -2.4199158e-01 ... -3.6186039e-02
7.1932510e-02 8.7262385e-02]
...
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-1.3326846e-01, -7.1100801e-02, -2.4199210e-01, ...,
1.1515099e-01, 2.8958965e-02, -1.1200016e-04],
[-1.3327029e-01, -7.1100898e-02, -2.4198994e-01, ...,
7.2522336e-03, -1.0089127e-01, 6.2182236e-02],
[-1.3326859e-01, -7.1101010e-02, -2.4199158e-01, ...,
-3.6186039e-02, 7.1932510e-02, 8.7262385e-02],
...,
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 128x0e+64x1e+32x2e
[[-1.3326846e-01 -7.1100801e-02 -2.4199210e-01 ... 1.1515099e-01
2.8958965e-02 -1.1200016e-04]
[-1.3327029e-01 -7.1100898e-02 -2.4198994e-01 ... 7.2522336e-03
-1.0089127e-01 6.2182236e-02]
[-1.3326859e-01 -7.1101010e-02 -2.4199158e-01 ... -3.6186039e-02
7.1932510e-02 8.7262385e-02]
...
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-1.3326846e-01, -7.1100801e-02, -2.4199210e-01, ...,
1.1515099e-01, 2.8958965e-02, -1.1200016e-04],
[-1.3327029e-01, -7.1100898e-02, -2.4198994e-01, ...,
7.2522336e-03, -1.0089127e-01, 6.2182236e-02],
[-1.3326859e-01, -7.1101010e-02, -2.4199158e-01, ...,
-3.6186039e-02, 7.1932510e-02, 8.7262385e-02],
...,
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, None,),), 'layer_1': layer_1input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-1.3326846e-01 -7.1100801e-02 -2.4199210e-01 ... 1.1515099e-01
2.8958965e-02 -1.1200016e-04]
[-1.3327029e-01 -7.1100898e-02 -2.4198994e-01 ... 7.2522336e-03
-1.0089127e-01 6.2182236e-02]
[-1.3326859e-01 -7.1101010e-02 -2.4199158e-01 ... -3.6186039e-02
7.1932510e-02 8.7262385e-02]
...
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-1.3326846e-01, -7.1100801e-02, -2.4199210e-01, ...,
1.1515099e-01, 2.8958965e-02, -1.1200016e-04],
[-1.3327029e-01, -7.1100898e-02, -2.4198994e-01, ...,
7.2522336e-03, -1.0089127e-01, 6.2182236e-02],
[-1.3326859e-01, -7.1101010e-02, -2.4199158e-01, ...,
-3.6186039e-02, 7.1932510e-02, 8.7262385e-02],
...,
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_species': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'interaction': interactioninput: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-1.3326846e-01 -7.1100801e-02 -2.4199210e-01 ... 1.1515099e-01
2.8958965e-02 -1.1200016e-04]
[-1.3327029e-01 -7.1100898e-02 -2.4198994e-01 ... 7.2522336e-03
-1.0089127e-01 6.2182236e-02]
[-1.3326859e-01 -7.1101010e-02 -2.4199158e-01 ... -3.6186039e-02
7.1932510e-02 8.7262385e-02]
...
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-1.3326846e-01, -7.1100801e-02, -2.4199210e-01, ...,
1.1515099e-01, 2.8958965e-02, -1.1200016e-04],
[-1.3327029e-01, -7.1100898e-02, -2.4198994e-01, ...,
7.2522336e-03, -1.0089127e-01, 6.2182236e-02],
[-1.3326859e-01, -7.1101010e-02, -2.4199158e-01, ...,
-3.6186039e-02, 7.1932510e-02, 8.7262385e-02],
...,
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'resid_adapter': resid_adapterinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[-1.3326846e-01 -7.1100801e-02 -2.4199210e-01 ... 1.1515099e-01
2.8958965e-02 -1.1200016e-04]
[-1.3327029e-01 -7.1100898e-02 -2.4198994e-01 ... 7.2522336e-03
-1.0089127e-01 6.2182236e-02]
[-1.3326859e-01 -7.1101010e-02 -2.4199158e-01 ... -3.6186039e-02
7.1932510e-02 8.7262385e-02]
...
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-1.3326846e-01, -7.1100801e-02, -2.4199210e-01, ...,
1.1515099e-01, 2.8958965e-02, -1.1200016e-04],
[-1.3327029e-01, -7.1100898e-02, -2.4198994e-01, ...,
7.2522336e-03, -1.0089127e-01, 6.2182236e-02],
[-1.3326859e-01, -7.1101010e-02, -2.4199158e-01, ...,
-3.6186039e-02, 7.1932510e-02, 8.7262385e-02],
...,
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 33792 parametersw[0,0] 128x0e,224x0e: Array([[-0.07596997, 0.27592677, -1.4187846 , ..., -3.4279747 ,
1.7853205 , 0.20697476],
[-0.5759112 , 0.37006366, -0.7312097 , ..., 0.05855327,
-1.455628 , 0.01047958],
[ 3.0478983 , 0.29383957, -1.444087 , ..., -2.9172626 ,
-2.151181 , 1.7908902 ],
...,
[ 1.0737331 , 1.1637673 , -1.0266234 , ..., 0.8998387 ,
0.19778317, -0.6234268 ],
[-0.60998833, 0.0140288 , 0.17401639, ..., -1.5538588 ,
2.0500894 , -0.22470145],
[ 0.97487503, 0.16422586, -1.996119 , ..., 1.6844265 ,
0.20047133, -1.0158596 ]], dtype=float32):Rendering array...
, w[1,1] 64x1e,64x1e: Array([[-1.8983171 , 1.1310133 , -1.0813652 , ..., 3.8598661 ,
2.1429558 , -3.254367 ],
[ 0.5758403 , 0.1438205 , -1.1729116 , ..., -0.40476117,
-1.7167196 , -0.3779419 ],
[-0.9746227 , 1.637837 , -0.05658302, ..., 0.01583426,
3.835662 , 2.9157832 ],
...,
[ 0.9743115 , 1.2317818 , 0.41286704, ..., 3.4449158 ,
-0.52608854, 0.23569065],
[ 0.68715185, -3.4482732 , 0.68070805, ..., -1.5814292 ,
-2.1625807 , -6.2308707 ],
[-1.7756596 , 2.637495 , -0.29488987, ..., 0.53851444,
1.3970186 , -6.2061367 ]], dtype=float32):Rendering array...
, w[2,2] 32x2e,32x2e: Array([[-3.4363009e-02, -4.7197115e-01, -1.3712782e+00, ...,
-1.2164356e+00, -1.1849614e+00, -6.1046684e-01],
[-7.4137789e-01, 3.9453822e-01, -1.5224578e+00, ...,
-6.9034678e-01, 1.0039064e+00, -1.2940993e+00],
[-6.5833054e-02, 1.0243961e+00, 1.2383620e+00, ...,
1.1483238e+00, 1.0708029e+00, 6.3221657e-01],
...,
[ 9.4188970e-01, -1.3368881e+00, 6.4393014e-02, ...,
-6.0346222e-01, -6.1310988e-02, -2.8410611e-01],
[-6.2800586e-01, 3.0434757e-01, 4.7315145e-01, ...,
-1.4627514e+00, -3.6217470e+00, 3.3898279e-03],
[ 3.7494395e+00, 3.9115545e-01, -8.2782209e-01, ...,
1.0386475e+00, -2.2443068e+00, 3.6211450e+00]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[ 4.2044035e-01 -1.8687807e+00 8.7565303e-01 ... -8.2087137e-02
2.1718955e-02 -1.6753847e-02]
[ 4.2044228e-01 -1.8687778e+00 8.7565035e-01 ... 1.5668305e-02
4.3450680e-02 -7.3193528e-02]
[ 4.2044118e-01 -1.8687811e+00 8.7565082e-01 ... -1.6910179e-02
-6.5169886e-02 -5.4383568e-02]
...
[-5.6051939e-45 9.8090893e-45 -1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.6051939e-45 9.8090893e-45 -1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.6051939e-45 9.8090893e-45 -1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 4.2044035e-01, -1.8687807e+00, 8.7565303e-01, ...,
-8.2087137e-02, 2.1718955e-02, -1.6753847e-02],
[ 4.2044228e-01, -1.8687778e+00, 8.7565035e-01, ...,
1.5668305e-02, 4.3450680e-02, -7.3193528e-02],
[ 4.2044118e-01, -1.8687811e+00, 8.7565082e-01, ...,
-1.6910179e-02, -6.5169886e-02, -5.4383568e-02],
...,
[-5.6051939e-45, 9.8090893e-45, -1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.6051939e-45, 9.8090893e-45, -1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.6051939e-45, 9.8090893e-45, -1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'SimpleInteraction_0': SimpleInteraction_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-1.3326846e-01 -7.1100801e-02 -2.4199210e-01 ... 1.1515099e-01
2.8958965e-02 -1.1200016e-04]
[-1.3327029e-01 -7.1100898e-02 -2.4198994e-01 ... 7.2522336e-03
-1.0089127e-01 6.2182236e-02]
[-1.3326859e-01 -7.1101010e-02 -2.4199158e-01 ... -3.6186039e-02
7.1932510e-02 8.7262385e-02]
...
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-1.3326846e-01, -7.1100801e-02, -2.4199210e-01, ...,
1.1515099e-01, 2.8958965e-02, -1.1200016e-04],
[-1.3327029e-01, -7.1100898e-02, -2.4198994e-01, ...,
7.2522336e-03, -1.0089127e-01, 6.2182236e-02],
[-1.3326859e-01, -7.1101010e-02, -2.4199158e-01, ...,
-3.6186039e-02, 7.1932510e-02, 8.7262385e-02],
...,
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'linear_intro': linear_introinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[-1.3326846e-01 -7.1100801e-02 -2.4199210e-01 ... 1.1515099e-01
2.8958965e-02 -1.1200016e-04]
[-1.3327029e-01 -7.1100898e-02 -2.4198994e-01 ... 7.2522336e-03
-1.0089127e-01 6.2182236e-02]
[-1.3326859e-01 -7.1101010e-02 -2.4199158e-01 ... -3.6186039e-02
7.1932510e-02 8.7262385e-02]
...
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -2.8025969e-45 2.8025969e-45 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-1.3326846e-01, -7.1100801e-02, -2.4199210e-01, ...,
1.1515099e-01, 2.8958965e-02, -1.1200016e-04],
[-1.3327029e-01, -7.1100898e-02, -2.4198994e-01, ...,
7.2522336e-03, -1.0089127e-01, 6.2182236e-02],
[-1.3326859e-01, -7.1101010e-02, -2.4199158e-01, ...,
-3.6186039e-02, 7.1932510e-02, 8.7262385e-02],
...,
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -2.8025969e-45, 2.8025969e-45, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 21504 parametersw[0,0] 128x0e,128x0e: Array([[-0.4241752 , 1.5493343 , 1.1070297 , ..., 0.7761027 ,
0.97573304, -0.42901313],
[-1.5982518 , -0.762294 , 0.925268 , ..., 2.184838 ,
2.0235624 , -0.91705096],
[ 0.15138112, -2.1470785 , -2.1204956 , ..., 0.18950811,
1.1807846 , 0.4766244 ],
...,
[-0.1753623 , -0.1232187 , 1.0429176 , ..., 4.9537115 ,
-1.2018416 , -1.9440458 ],
[ 0.53979814, 0.6425637 , -1.3816024 , ..., -0.44571406,
0.14153725, 0.9647347 ],
[ 2.1930974 , -0.6061898 , -0.9843441 , ..., 2.3418922 ,
2.5735004 , -2.294102 ]], dtype=float32):Rendering array...
, w[1,1] 64x1e,64x1e: Array([[ 1.3482215 , -3.131865 , -3.503224 , ..., 2.234558 ,
4.6431923 , -0.61837673],
[ 7.4621325 , -2.6024756 , -3.024557 , ..., 0.1146721 ,
-2.6666873 , -4.892023 ],
[-1.1567578 , 2.3914483 , -3.5104413 , ..., -2.923205 ,
-2.8599203 , 1.8045309 ],
...,
[ 3.134736 , -1.4929918 , 4.686314 , ..., -3.973341 ,
-8.705682 , 2.4547012 ],
[ 0.9738718 , -1.421875 , -1.2021968 , ..., 2.3651967 ,
-4.870255 , 1.5145608 ],
[ 3.0686798 , 4.702751 , -2.2753363 , ..., 2.1597495 ,
2.648089 , -0.3652111 ]], dtype=float32):Rendering array...
, w[2,2] 32x2e,32x2e: Array([[-3.3317432 , 1.6920732 , 0.44324556, ..., 4.037474 ,
-1.1616123 , 3.2893333 ],
[ 0.64936876, -3.7995343 , -0.47002053, ..., 3.9982092 ,
3.9305694 , 2.3129532 ],
[-1.3567086 , 6.0881386 , -0.23467521, ..., 2.5168328 ,
2.0815182 , 3.4210553 ],
...,
[-2.493092 , -2.0121279 , 1.7890359 , ..., -2.482454 ,
-2.4887905 , 6.641017 ],
[ 2.6620302 , 3.5397139 , -1.93738 , ..., -3.9678414 ,
-0.5936509 , -5.3279395 ],
[ 0.86180013, 2.0567636 , 1.5637033 , ..., 0.9133283 ,
0.8062022 , -1.2511878 ]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[-2.13428593e+00 3.37838501e-01 -9.30966735e-01 ... -7.76204944e-01
1.25402004e-01 -1.16331518e-01]
[-2.13428164e+00 3.37842315e-01 -9.30969596e-01 ... 1.15517884e-01
4.69075441e-01 -6.31168485e-01]
[-2.13428569e+00 3.37839276e-01 -9.30969417e-01 ... -7.25841075e-02
-5.94482124e-01 -5.22566557e-01]
...
[ 1.96181785e-44 -7.00649232e-45 -5.60519386e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 1.96181785e-44 -7.00649232e-45 -5.60519386e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 1.96181785e-44 -7.00649232e-45 -5.60519386e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-2.13428593e+00, 3.37838501e-01, -9.30966735e-01, ...,
-7.76204944e-01, 1.25402004e-01, -1.16331518e-01],
[-2.13428164e+00, 3.37842315e-01, -9.30969596e-01, ...,
1.15517884e-01, 4.69075441e-01, -6.31168485e-01],
[-2.13428569e+00, 3.37839276e-01, -9.30969417e-01, ...,
-7.25841075e-02, -5.94482124e-01, -5.22566557e-01],
...,
[ 1.96181785e-44, -7.00649232e-45, -5.60519386e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 1.96181785e-44, -7.00649232e-45, -5.60519386e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 1.96181785e-44, -7.00649232e-45, -5.60519386e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'SevenNetConv_0': SevenNetConv_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-2.13428593e+00 3.37838501e-01 -9.30966735e-01 ... -7.76204944e-01
1.25402004e-01 -1.16331518e-01]
[-2.13428164e+00 3.37842315e-01 -9.30969596e-01 ... 1.15517884e-01
4.69075441e-01 -6.31168485e-01]
[-2.13428569e+00 3.37839276e-01 -9.30969417e-01 ... -7.25841075e-02
-5.94482124e-01 -5.22566557e-01]
...
[ 1.96181785e-44 -7.00649232e-45 -5.60519386e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 1.96181785e-44 -7.00649232e-45 -5.60519386e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[ 1.96181785e-44 -7.00649232e-45 -5.60519386e-45 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[-2.13428593e+00, 3.37838501e-01, -9.30966735e-01, ...,
-7.76204944e-01, 1.25402004e-01, -1.16331518e-01],
[-2.13428164e+00, 3.37842315e-01, -9.30969596e-01, ...,
1.15517884e-01, 4.69075441e-01, -6.31168485e-01],
[-2.13428569e+00, 3.37839276e-01, -9.30969417e-01, ...,
-7.25841075e-02, -5.94482124e-01, -5.22566557e-01],
...,
[ 1.96181785e-44, -7.00649232e-45, -5.60519386e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 1.96181785e-44, -7.00649232e-45, -5.60519386e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[ 1.96181785e-44, -7.00649232e-45, -5.60519386e-45, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'LazyInMLP_0': LazyInMLP_0input: 'x': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,,}children: 'Dense_0': Dense_0input: 'inputs': Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32):Rendering array...
,}params: 7680 parameterskernel: Array([[ 4.052882 , -0.22530393, 1.4029229 , ..., 1.0817723 ,
-0.03070471, 1.6746305 ],
[ 5.135906 , -1.0228736 , 3.1861758 , ..., 2.607143 ,
4.6757793 , 4.5255265 ],
[ 2.4167147 , -0.24284168, 1.9592719 , ..., 1.1998506 ,
3.568256 , 3.412591 ],
...,
[ 0.69481957, -1.947504 , 2.3535092 , ..., -5.6909156 ,
1.383382 , 5.6895742 ],
[ 0.2564721 , -1.988878 , -0.11093269, ..., -3.8434715 ,
1.7036772 , 4.353121 ],
[-0.0734681 , -0.701135 , -0.6696881 , ..., -1.1655374 ,
-0.3284213 , 0.2726876 ]], dtype=float32):Rendering array...
,}output: Array([[[ 1.82870078e+00, -6.62220716e-01, -1.64561242e-01, ...,
1.97314012e+00, 1.62335014e+00, -5.23620546e-02],
[ 1.33937311e+00, -1.02675724e+00, 6.91671014e-01, ...,
-4.61365968e-01, 1.07399237e+00, 1.51709604e+00],
[ 6.09271169e-01, 1.04536144e-02, 5.35353661e-01, ...,
-7.54954740e-02, -5.21018088e-01, 2.83327103e-01],
...,
[ 1.56871751e-02, -1.37338126e-02, -3.05151008e-03, ...,
-2.59463191e-02, -3.64792608e-02, 2.14571245e-02],
[ 1.44898985e-02, -1.24931475e-02, -2.84474064e-03, ...,
-2.36894805e-02, -3.39862257e-02, 1.89456847e-02],
[ 1.20376768e-02, -1.00243539e-02, -2.40167952e-03, ...,
-1.91756040e-02, -2.87451912e-02, 1.41618997e-02]],
[[ 1.82870078e+00, -6.62220716e-01, -1.64561242e-01, ...,
1.97314012e+00, 1.62335014e+00, -5.23620546e-02],
[ 1.33937335e+00, -1.02675760e+00, 6.91670239e-01, ...,
-4.61365283e-01, 1.07399285e+00, 1.51709545e+00],
[ 6.09269798e-01, 1.04545997e-02, 5.35351455e-01, ...,
-7.54947066e-02, -5.21019697e-01, 2.83325016e-01],
...,
[ 1.56871751e-02, -1.37338126e-02, -3.05151008e-03, ...,
-2.59463191e-02, -3.64792608e-02, 2.14571245e-02],
[ 1.44897765e-02, -1.24929687e-02, -2.84479652e-03, ...,
-2.36891042e-02, -3.39861736e-02, 1.89452395e-02],
[ 1.20377662e-02, -1.00244759e-02, -2.40168930e-03, ...,
-1.91758256e-02, -2.87453085e-02, 1.41621530e-02]],
[[ 1.82870054e+00, -6.62221074e-01, -1.64561242e-01, ...,
1.97313941e+00, 1.62335014e+00, -5.23614772e-02],
[ 1.33937347e+00, -1.02675784e+00, 6.91670120e-01, ...,
-4.61365342e-01, 1.07399356e+00, 1.51709569e+00],
[ 6.09271049e-01, 1.04534263e-02, 5.35353482e-01, ...,
-7.54962489e-02, -5.21018147e-01, 2.83327669e-01],
...,
[ 1.56872347e-02, -1.37338489e-02, -3.05157690e-03, ...,
-2.59463526e-02, -3.64794694e-02, 2.14571767e-02],
[ 1.44898985e-02, -1.24931475e-02, -2.84474064e-03, ...,
-2.36894805e-02, -3.39862257e-02, 1.89456847e-02],
[ 1.20376768e-02, -1.00243539e-02, -2.40167952e-03, ...,
-1.91756040e-02, -2.87451912e-02, 1.41618997e-02]],
...,
[[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
...,
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01]],
[[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
...,
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01]],
[[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
...,
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01]]], dtype=float32):Rendering array...
,),}output: 960x0e
[[[ 1.82870078e+00 -6.62220716e-01 -1.64561242e-01 ... 1.97314012e+00
1.62335014e+00 -5.23620546e-02]
[ 1.33937311e+00 -1.02675724e+00 6.91671014e-01 ... -4.61365968e-01
1.07399237e+00 1.51709604e+00]
[ 6.09271169e-01 1.04536144e-02 5.35353661e-01 ... -7.54954740e-02
-5.21018088e-01 2.83327103e-01]
...
[ 1.56871751e-02 -1.37338126e-02 -3.05151008e-03 ... -2.59463191e-02
-3.64792608e-02 2.14571245e-02]
[ 1.44898985e-02 -1.24931475e-02 -2.84474064e-03 ... -2.36894805e-02
-3.39862257e-02 1.89456847e-02]
[ 1.20376768e-02 -1.00243539e-02 -2.40167952e-03 ... -1.91756040e-02
-2.87451912e-02 1.41618997e-02]]
[[ 1.82870078e+00 -6.62220716e-01 -1.64561242e-01 ... 1.97314012e+00
1.62335014e+00 -5.23620546e-02]
[ 1.33937335e+00 -1.02675760e+00 6.91670239e-01 ... -4.61365283e-01
1.07399285e+00 1.51709545e+00]
[ 6.09269798e-01 1.04545997e-02 5.35351455e-01 ... -7.54947066e-02
-5.21019697e-01 2.83325016e-01]
...
[ 1.56871751e-02 -1.37338126e-02 -3.05151008e-03 ... -2.59463191e-02
-3.64792608e-02 2.14571245e-02]
[ 1.44897765e-02 -1.24929687e-02 -2.84479652e-03 ... -2.36891042e-02
-3.39861736e-02 1.89452395e-02]
[ 1.20377662e-02 -1.00244759e-02 -2.40168930e-03 ... -1.91758256e-02
-2.87453085e-02 1.41621530e-02]]
[[ 1.82870054e+00 -6.62221074e-01 -1.64561242e-01 ... 1.97313941e+00
1.62335014e+00 -5.23614772e-02]
[ 1.33937347e+00 -1.02675784e+00 6.91670120e-01 ... -4.61365342e-01
1.07399356e+00 1.51709569e+00]
[ 6.09271049e-01 1.04534263e-02 5.35353482e-01 ... -7.54962489e-02
-5.21018147e-01 2.83327669e-01]
...
[ 1.56872347e-02 -1.37338489e-02 -3.05157690e-03 ... -2.59463526e-02
-3.64794694e-02 2.14571767e-02]
[ 1.44898985e-02 -1.24931475e-02 -2.84474064e-03 ... -2.36894805e-02
-3.39862257e-02 1.89456847e-02]
[ 1.20376768e-02 -1.00243539e-02 -2.40167952e-03 ... -1.91756040e-02
-2.87451912e-02 1.41618997e-02]]
...
[[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
...
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]]
[[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
...
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]]
[[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
...
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]
[ 7.42023087e+00 -7.44910336e+00 9.72869301e+00 ... -2.11137772e+01
9.08848000e+00 3.09475784e+01]]]key=0,): Array([[[ 1.82870078e+00, -6.62220716e-01, -1.64561242e-01, ...,
1.97314012e+00, 1.62335014e+00, -5.23620546e-02],
[ 1.33937311e+00, -1.02675724e+00, 6.91671014e-01, ...,
-4.61365968e-01, 1.07399237e+00, 1.51709604e+00],
[ 6.09271169e-01, 1.04536144e-02, 5.35353661e-01, ...,
-7.54954740e-02, -5.21018088e-01, 2.83327103e-01],
...,
[ 1.56871751e-02, -1.37338126e-02, -3.05151008e-03, ...,
-2.59463191e-02, -3.64792608e-02, 2.14571245e-02],
[ 1.44898985e-02, -1.24931475e-02, -2.84474064e-03, ...,
-2.36894805e-02, -3.39862257e-02, 1.89456847e-02],
[ 1.20376768e-02, -1.00243539e-02, -2.40167952e-03, ...,
-1.91756040e-02, -2.87451912e-02, 1.41618997e-02]],
[[ 1.82870078e+00, -6.62220716e-01, -1.64561242e-01, ...,
1.97314012e+00, 1.62335014e+00, -5.23620546e-02],
[ 1.33937335e+00, -1.02675760e+00, 6.91670239e-01, ...,
-4.61365283e-01, 1.07399285e+00, 1.51709545e+00],
[ 6.09269798e-01, 1.04545997e-02, 5.35351455e-01, ...,
-7.54947066e-02, -5.21019697e-01, 2.83325016e-01],
...,
[ 1.56871751e-02, -1.37338126e-02, -3.05151008e-03, ...,
-2.59463191e-02, -3.64792608e-02, 2.14571245e-02],
[ 1.44897765e-02, -1.24929687e-02, -2.84479652e-03, ...,
-2.36891042e-02, -3.39861736e-02, 1.89452395e-02],
[ 1.20377662e-02, -1.00244759e-02, -2.40168930e-03, ...,
-1.91758256e-02, -2.87453085e-02, 1.41621530e-02]],
[[ 1.82870054e+00, -6.62221074e-01, -1.64561242e-01, ...,
1.97313941e+00, 1.62335014e+00, -5.23614772e-02],
[ 1.33937347e+00, -1.02675784e+00, 6.91670120e-01, ...,
-4.61365342e-01, 1.07399356e+00, 1.51709569e+00],
[ 6.09271049e-01, 1.04534263e-02, 5.35353482e-01, ...,
-7.54962489e-02, -5.21018147e-01, 2.83327669e-01],
...,
[ 1.56872347e-02, -1.37338489e-02, -3.05157690e-03, ...,
-2.59463526e-02, -3.64794694e-02, 2.14571767e-02],
[ 1.44898985e-02, -1.24931475e-02, -2.84474064e-03, ...,
-2.36894805e-02, -3.39862257e-02, 1.89456847e-02],
[ 1.20376768e-02, -1.00243539e-02, -2.40167952e-03, ...,
-1.91756040e-02, -2.87451912e-02, 1.41618997e-02]],
...,
[[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
...,
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01]],
[[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
...,
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01]],
[[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
...,
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01],
[ 7.42023087e+00, -7.44910336e+00, 9.72869301e+00, ...,
-2.11137772e+01, 9.08848000e+00, 3.09475784e+01]]], dtype=float32), :Rendering array...
,,),}output: 224x0e+384x1e+352x2e
[[-0.257615 0.05476849 -0.02814175 ... 0.00187587 -0.01420697
0.00677627]
[-0.25761732 0.05475245 -0.02813954 ... -0.00649269 0.0086561
0.01160528]
[-0.25761712 0.05475226 -0.0281394 ... 0.01481542 0.00558733
-0.00068669]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.257615 , 0.05476849, -0.02814175, ..., 0.00187587,
-0.01420697, 0.00677627],
[-0.25761732, 0.05475245, -0.02813954, ..., -0.00649269,
0.0086561 , 0.01160528],
[-0.25761712, 0.05475226, -0.0281394 , ..., 0.01481542,
0.00558733, -0.00068669],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'linear_outro': linear_outroinput: 'weights_or_input': 224x0e+384x1e+352x2e
[[-0.257615 0.05476849 -0.02814175 ... 0.00187587 -0.01420697
0.00677627]
[-0.25761732 0.05475245 -0.02813954 ... -0.00649269 0.0086561
0.01160528]
[-0.25761712 0.05475226 -0.0281394 ... 0.01481542 0.00558733
-0.00068669]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.257615 , 0.05476849, -0.02814175, ..., 0.00187587,
-0.01420697, 0.00677627],
[-0.25761732, 0.05475245, -0.02813954, ..., -0.00649269,
0.0086561 , 0.01160528],
[-0.25761712, 0.05475226, -0.0281394 , ..., 0.01481542,
0.00558733, -0.00068669],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 86016 parametersw[0,0] 224x0e,224x0e: Array([[ 3.6113713 , -1.1181697 , 0.03881168, ..., -2.1805305 ,
0.64018303, 1.2658951 ],
[-3.7488034 , -0.92508084, -1.101863 , ..., 0.8667244 ,
-3.71324 , -0.2927172 ],
[ 1.1253558 , 1.2100322 , 0.13155553, ..., -2.042996 ,
-2.523338 , 3.6498551 ],
...,
[ 3.6797526 , 3.550883 , 1.4122659 , ..., 2.826467 ,
1.8865697 , -1.0979416 ],
[ 0.4086895 , 0.00904947, -0.63308686, ..., -0.81488013,
3.6393635 , -3.7455666 ],
[ 1.8093982 , -3.068014 , 0.901131 , ..., -1.4847515 ,
-0.36652914, 1.6428287 ]], dtype=float32):Rendering array...
, w[1,1] 384x1e,64x1e: Array([[-0.79561204, -0.30382457, 0.0335411 , ..., -1.2693908 ,
0.37836167, 2.9927666 ],
[-1.455988 , -1.9610479 , -0.44855782, ..., 0.03941472,
0.62473786, 4.7411556 ],
[ 0.7098443 , 0.6184685 , -2.0183759 , ..., -5.9243107 ,
-4.4614496 , 2.7931871 ],
...,
[-0.7004779 , 1.4356883 , -0.9977043 , ..., -0.51720756,
0.15333058, -1.0994623 ],
[ 0.77050114, 1.6966777 , -1.5533303 , ..., -1.0010715 ,
-0.4932081 , 0.49124607],
[-1.25783 , -0.6229767 , 0.94102746, ..., -0.6150069 ,
0.45159495, 0.16357416]], dtype=float32):Rendering array...
, w[2,2] 352x2e,32x2e: Array([[ 1.9871085 , -3.3893056 , -0.04077484, ..., 0.21905695,
0.7577846 , -0.5522776 ],
[ -0.5850577 , 0.19033383, -2.503004 , ..., 1.1635637 ,
1.7451153 , -3.0206232 ],
[ 1.2223674 , -1.2091054 , 0.39926517, ..., 2.9988463 ,
-1.7507242 , -3.1746583 ],
...,
[ 0.5759754 , -0.16012855, 3.1206794 , ..., 3.6077344 ,
-11.050401 , 0.70109856],
[ 2.6171324 , 1.2029477 , 5.376402 , ..., 0.8496376 ,
-0.45670396, 1.9353403 ],
[ -0.88361245, 5.8734903 , -1.6447512 , ..., 3.8160083 ,
2.0072403 , -1.1579607 ]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[-0.8229257 -1.3910886 -1.6463009 ... -0.0391192 -0.07258879
0.06422254]
[-0.8229156 -1.3910838 -1.6463012 ... -0.0078299 0.09338453
0.04611334]
[-0.8229172 -1.3910754 -1.6462958 ... 0.10098165 -0.020834
-0.01664447]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.8229257 , -1.3910886 , -1.6463009 , ..., -0.0391192 ,
-0.07258879, 0.06422254],
[-0.8229156 , -1.3910838 , -1.6463012 , ..., -0.0078299 ,
0.09338453, 0.04611334],
[-0.8229172 , -1.3910754 , -1.6462958 , ..., 0.10098165,
-0.020834 , -0.01664447],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[-0.8229257 -1.3910886 -1.6463009 ... -0.0391192 -0.07258879
0.06422254]
[-0.8229156 -1.3910838 -1.6463012 ... -0.0078299 0.09338453
0.04611334]
[-0.8229172 -1.3910754 -1.6462958 ... 0.10098165 -0.020834
-0.01664447]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.8229257 , -1.3910886 , -1.6463009 , ..., -0.0391192 ,
-0.07258879, 0.06422254],
[-0.8229156 , -1.3910838 , -1.6463012 , ..., -0.0078299 ,
0.09338453, 0.04611334],
[-0.8229172 , -1.3910754 , -1.6462958 , ..., 0.10098165,
-0.020834 , -0.01664447],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[-4.0248534e-01 -3.2598693e+00 -7.7064788e-01 ... -1.2120634e-01
-5.0869837e-02 4.7468700e-02]
[-4.0247333e-01 -3.2598615e+00 -7.7065080e-01 ... 7.8384038e-03
1.3683522e-01 -2.7080189e-02]
[-4.0247604e-01 -3.2598565e+00 -7.7064496e-01 ... 8.4071465e-02
-8.6003892e-02 -7.1028039e-02]
...
[-5.6051939e-45 9.8090893e-45 -1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.6051939e-45 9.8090893e-45 -1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.6051939e-45 9.8090893e-45 -1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-4.0248534e-01, -3.2598693e+00, -7.7064788e-01, ...,
-1.2120634e-01, -5.0869837e-02, 4.7468700e-02],
[-4.0247333e-01, -3.2598615e+00, -7.7065080e-01, ...,
7.8384038e-03, 1.3683522e-01, -2.7080189e-02],
[-4.0247604e-01, -3.2598565e+00, -7.7064496e-01, ...,
8.4071465e-02, -8.6003892e-02, -7.1028039e-02],
...,
[-5.6051939e-45, 9.8090893e-45, -1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.6051939e-45, 9.8090893e-45, -1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.6051939e-45, 9.8090893e-45, -1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'self_connection': self_connectioninput: 'node_feats': 224x0e+64x1e+32x2e
[[-4.0248534e-01 -3.2598693e+00 -7.7064788e-01 ... -1.2120634e-01
-5.0869837e-02 4.7468700e-02]
[-4.0247333e-01 -3.2598615e+00 -7.7065080e-01 ... 7.8384038e-03
1.3683522e-01 -2.7080189e-02]
[-4.0247604e-01 -3.2598565e+00 -7.7064496e-01 ... 8.4071465e-02
-8.6003892e-02 -7.1028039e-02]
...
[-5.6051939e-45 9.8090893e-45 -1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.6051939e-45 9.8090893e-45 -1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.6051939e-45 9.8090893e-45 -1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-4.0248534e-01, -3.2598693e+00, -7.7064788e-01, ...,
-1.2120634e-01, -5.0869837e-02, 4.7468700e-02],
[-4.0247333e-01, -3.2598615e+00, -7.7065080e-01, ...,
7.8384038e-03, 1.3683522e-01, -2.7080189e-02],
[-4.0247604e-01, -3.2598565e+00, -7.7064496e-01, ...,
8.4071465e-02, -8.6003892e-02, -7.1028039e-02],
...,
[-5.6051939e-45, 9.8090893e-45, -1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.6051939e-45, 9.8090893e-45, -1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.6051939e-45, 9.8090893e-45, -1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_specie': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[-2.7039984e-01 -2.0207627e-01 -4.0872300e-01 ... 3.7595402e-02
1.5778646e-02 -1.4723691e-02]
[-2.7039370e-01 -2.0207733e-01 -4.0872368e-01 ... -2.4312863e-03
-4.2443026e-02 8.3996300e-03]
[-2.7039510e-01 -2.0207798e-01 -4.0872225e-01 ... -2.6077012e-02
2.6676405e-02 2.2031246e-02]
...
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.7039984e-01, -2.0207627e-01, -4.0872300e-01, ...,
3.7595402e-02, 1.5778646e-02, -1.4723691e-02],
[-2.7039370e-01, -2.0207733e-01, -4.0872368e-01, ...,
-2.4312863e-03, -4.2443026e-02, 8.3996300e-03],
[-2.7039510e-01, -2.0207798e-01, -4.0872225e-01, ...,
-2.6077012e-02, 2.6676405e-02, 2.2031246e-02],
...,
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 128x0e+64x1e+32x2e
[[-2.7039984e-01 -2.0207627e-01 -4.0872300e-01 ... 3.7595402e-02
1.5778646e-02 -1.4723691e-02]
[-2.7039370e-01 -2.0207733e-01 -4.0872368e-01 ... -2.4312863e-03
-4.2443026e-02 8.3996300e-03]
[-2.7039510e-01 -2.0207798e-01 -4.0872225e-01 ... -2.6077012e-02
2.6676405e-02 2.2031246e-02]
...
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.7039984e-01, -2.0207627e-01, -4.0872300e-01, ...,
3.7595402e-02, 1.5778646e-02, -1.4723691e-02],
[-2.7039370e-01, -2.0207733e-01, -4.0872368e-01, ...,
-2.4312863e-03, -4.2443026e-02, 8.3996300e-03],
[-2.7039510e-01, -2.0207798e-01, -4.0872225e-01, ...,
-2.6077012e-02, 2.6676405e-02, 2.2031246e-02],
...,
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, None,),), 'layer_2': layer_2input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-2.7039984e-01 -2.0207627e-01 -4.0872300e-01 ... 3.7595402e-02
1.5778646e-02 -1.4723691e-02]
[-2.7039370e-01 -2.0207733e-01 -4.0872368e-01 ... -2.4312863e-03
-4.2443026e-02 8.3996300e-03]
[-2.7039510e-01 -2.0207798e-01 -4.0872225e-01 ... -2.6077012e-02
2.6676405e-02 2.2031246e-02]
...
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.7039984e-01, -2.0207627e-01, -4.0872300e-01, ...,
3.7595402e-02, 1.5778646e-02, -1.4723691e-02],
[-2.7039370e-01, -2.0207733e-01, -4.0872368e-01, ...,
-2.4312863e-03, -4.2443026e-02, 8.3996300e-03],
[-2.7039510e-01, -2.0207798e-01, -4.0872225e-01, ...,
-2.6077012e-02, 2.6676405e-02, 2.2031246e-02],
...,
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_species': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'interaction': interactioninput: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-2.7039984e-01 -2.0207627e-01 -4.0872300e-01 ... 3.7595402e-02
1.5778646e-02 -1.4723691e-02]
[-2.7039370e-01 -2.0207733e-01 -4.0872368e-01 ... -2.4312863e-03
-4.2443026e-02 8.3996300e-03]
[-2.7039510e-01 -2.0207798e-01 -4.0872225e-01 ... -2.6077012e-02
2.6676405e-02 2.2031246e-02]
...
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.7039984e-01, -2.0207627e-01, -4.0872300e-01, ...,
3.7595402e-02, 1.5778646e-02, -1.4723691e-02],
[-2.7039370e-01, -2.0207733e-01, -4.0872368e-01, ...,
-2.4312863e-03, -4.2443026e-02, 8.3996300e-03],
[-2.7039510e-01, -2.0207798e-01, -4.0872225e-01, ...,
-2.6077012e-02, 2.6676405e-02, 2.2031246e-02],
...,
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'resid_adapter': resid_adapterinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[-2.7039984e-01 -2.0207627e-01 -4.0872300e-01 ... 3.7595402e-02
1.5778646e-02 -1.4723691e-02]
[-2.7039370e-01 -2.0207733e-01 -4.0872368e-01 ... -2.4312863e-03
-4.2443026e-02 8.3996300e-03]
[-2.7039510e-01 -2.0207798e-01 -4.0872225e-01 ... -2.6077012e-02
2.6676405e-02 2.2031246e-02]
...
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.7039984e-01, -2.0207627e-01, -4.0872300e-01, ...,
3.7595402e-02, 1.5778646e-02, -1.4723691e-02],
[-2.7039370e-01, -2.0207733e-01, -4.0872368e-01, ...,
-2.4312863e-03, -4.2443026e-02, 8.3996300e-03],
[-2.7039510e-01, -2.0207798e-01, -4.0872225e-01, ...,
-2.6077012e-02, 2.6676405e-02, 2.2031246e-02],
...,
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 33792 parametersw[0,0] 128x0e,224x0e: Array([[-0.2698589 , 0.44674757, -0.31250012, ..., -3.6976795 ,
-2.4428482 , -1.1022315 ],
[-0.19646236, -0.9834637 , 0.63143486, ..., -0.05126027,
1.1076013 , 1.7861416 ],
[ 0.48316053, 2.9596577 , 0.12952845, ..., -0.56942916,
-0.40692878, -0.22096917],
...,
[-3.9606133 , 0.03599317, 1.4394854 , ..., 0.19181603,
-1.2654991 , -1.2876765 ],
[-2.0281868 , -0.13632914, -3.4674752 , ..., -0.2869467 ,
-0.17545098, -1.5250418 ],
[ 4.2003765 , 0.12415113, -0.42462096, ..., 2.0245545 ,
-2.225135 , 2.1616156 ]], dtype=float32):Rendering array...
, w[1,1] 64x1e,64x1e: Array([[ 2.2523465 , -1.2124305 , 0.4407775 , ..., -0.11525499,
0.6420827 , -1.1333939 ],
[-2.3259945 , -2.6467135 , -0.9030307 , ..., -3.7320697 ,
2.6729276 , -3.3454947 ],
[ 3.0255785 , 0.5180181 , 2.7732878 , ..., 1.524265 ,
-2.1999319 , -0.20321423],
...,
[-0.57588875, -0.53166825, 0.2651183 , ..., 2.953936 ,
-1.5758833 , -2.0478845 ],
[-2.9871452 , 5.09833 , -0.8852749 , ..., -1.3664873 ,
1.2529833 , 1.8465164 ],
[-4.4754567 , 3.4820943 , -2.7085273 , ..., -2.4177372 ,
-0.8317587 , -0.06291667]], dtype=float32):Rendering array...
, w[2,2] 32x2e,32x2e: Array([[-0.42950523, 0.9804102 , -0.34976012, ..., -3.4437647 ,
-3.3836908 , -0.8078704 ],
[-1.6786072 , -1.2309582 , 0.22875902, ..., -0.72541004,
1.2473128 , 0.3433992 ],
[-0.87624604, 1.8254704 , -3.3011746 , ..., -1.3931991 ,
-1.4904697 , 2.3584936 ],
...,
[ 1.6216624 , -1.4492549 , -0.44740498, ..., 0.48102644,
-1.2849988 , 0.5881696 ],
[-0.59220976, -0.5082739 , -1.5561093 , ..., -0.5740662 ,
-0.3018164 , 1.8806973 ],
[-1.3057946 , -0.49188724, 0.6209479 , ..., -0.8584203 ,
-0.4544655 , 0.9636952 ]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[-9.5111948e-01 -1.9130801e+00 -1.0549632e+00 ... 7.8853451e-02
-4.0722176e-02 -1.4002653e-02]
[-9.5111811e-01 -1.9130800e+00 -1.0549588e+00 ... -4.9504988e-02
-4.4822525e-02 6.0099024e-02]
[-9.5112294e-01 -1.9130623e+00 -1.0549576e+00 ... 1.1610745e-02
8.5578546e-02 2.4836555e-02]
...
[ 5.6051939e-45 1.4012985e-45 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 5.6051939e-45 1.4012985e-45 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 5.6051939e-45 1.4012985e-45 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-9.5111948e-01, -1.9130801e+00, -1.0549632e+00, ...,
7.8853451e-02, -4.0722176e-02, -1.4002653e-02],
[-9.5111811e-01, -1.9130800e+00, -1.0549588e+00, ...,
-4.9504988e-02, -4.4822525e-02, 6.0099024e-02],
[-9.5112294e-01, -1.9130623e+00, -1.0549576e+00, ...,
1.1610745e-02, 8.5578546e-02, 2.4836555e-02],
...,
[ 5.6051939e-45, 1.4012985e-45, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 5.6051939e-45, 1.4012985e-45, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 5.6051939e-45, 1.4012985e-45, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'SimpleInteraction_0': SimpleInteraction_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-2.7039984e-01 -2.0207627e-01 -4.0872300e-01 ... 3.7595402e-02
1.5778646e-02 -1.4723691e-02]
[-2.7039370e-01 -2.0207733e-01 -4.0872368e-01 ... -2.4312863e-03
-4.2443026e-02 8.3996300e-03]
[-2.7039510e-01 -2.0207798e-01 -4.0872225e-01 ... -2.6077012e-02
2.6676405e-02 2.2031246e-02]
...
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.7039984e-01, -2.0207627e-01, -4.0872300e-01, ...,
3.7595402e-02, 1.5778646e-02, -1.4723691e-02],
[-2.7039370e-01, -2.0207733e-01, -4.0872368e-01, ...,
-2.4312863e-03, -4.2443026e-02, 8.3996300e-03],
[-2.7039510e-01, -2.0207798e-01, -4.0872225e-01, ...,
-2.6077012e-02, 2.6676405e-02, 2.2031246e-02],
...,
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'linear_intro': linear_introinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[-2.7039984e-01 -2.0207627e-01 -4.0872300e-01 ... 3.7595402e-02
1.5778646e-02 -1.4723691e-02]
[-2.7039370e-01 -2.0207733e-01 -4.0872368e-01 ... -2.4312863e-03
-4.2443026e-02 8.3996300e-03]
[-2.7039510e-01 -2.0207798e-01 -4.0872225e-01 ... -2.6077012e-02
2.6676405e-02 2.2031246e-02]
...
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-4.2038954e-45 9.8090893e-45 -1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-2.7039984e-01, -2.0207627e-01, -4.0872300e-01, ...,
3.7595402e-02, 1.5778646e-02, -1.4723691e-02],
[-2.7039370e-01, -2.0207733e-01, -4.0872368e-01, ...,
-2.4312863e-03, -4.2443026e-02, 8.3996300e-03],
[-2.7039510e-01, -2.0207798e-01, -4.0872225e-01, ...,
-2.6077012e-02, 2.6676405e-02, 2.2031246e-02],
...,
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-4.2038954e-45, 9.8090893e-45, -1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 21504 parametersw[0,0] 128x0e,128x0e: Array([[-2.1586008 , -1.6786722 , 0.5165552 , ..., 0.70764744,
-3.3874962 , -0.5979072 ],
[ 1.4567249 , -0.78757703, 0.14725283, ..., 3.9408135 ,
-2.4566958 , 2.719328 ],
[ 1.5687432 , -1.134582 , 0.3726073 , ..., -0.5147848 ,
-2.6269054 , 1.254605 ],
...,
[-4.5462236 , -2.068843 , -0.07038818, ..., -2.0033019 ,
0.72902334, -1.1020043 ],
[ 0.6327289 , -3.9143023 , 0.30573696, ..., -1.8584297 ,
2.3847718 , 1.774803 ],
[ 0.5761787 , 2.6509273 , 1.4255936 , ..., 1.262279 ,
2.7701647 , -2.3315237 ]], dtype=float32):Rendering array...
, w[1,1] 64x1e,64x1e: Array([[ 0.6084777 , -0.66246223, 6.619954 , ..., -0.9813953 ,
1.444019 , 1.6443533 ],
[-1.2975378 , 2.1629822 , -2.5576208 , ..., -2.1381264 ,
0.77533215, -1.0800792 ],
[ 1.0625116 , -2.1124494 , -0.28465724, ..., -0.13889353,
-2.429628 , -3.6430233 ],
...,
[-0.583007 , 3.0251534 , 3.7068667 , ..., 2.6721835 ,
1.8640645 , 3.0045564 ],
[ 2.132253 , 5.790305 , 1.7161772 , ..., -1.8887897 ,
1.0866327 , 0.86480767],
[-0.6644205 , 4.505556 , 2.7787488 , ..., -0.5326051 ,
3.594365 , -9.192757 ]], dtype=float32):Rendering array...
, w[2,2] 32x2e,32x2e: Array([[-4.14843 , -2.5014877 , -1.4684834 , ..., -5.8136964 ,
0.8666029 , 3.7350478 ],
[-3.7453384 , 1.0307775 , -3.6572578 , ..., 1.1598345 ,
-0.38757676, 0.70694864],
[-1.8403573 , 0.3684081 , 0.52219 , ..., -1.808346 ,
-0.20554003, 2.7274349 ],
...,
[-0.22899586, 5.4061036 , 2.9266653 , ..., 1.767967 ,
3.704977 , -4.4892683 ],
[-0.06069649, 3.433467 , -3.772579 , ..., 1.0408311 ,
0.47197062, -0.97904927],
[ 5.2918415 , -5.065796 , -5.8402925 , ..., -2.3323772 ,
-2.5990508 , 7.3229866 ]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[ 7.8903055e-01 3.0284563e-01 1.7624679e+00 ... 1.6439064e-01
-4.7409746e-01 1.6958097e-01]
[ 7.8903079e-01 3.0283996e-01 1.7624480e+00 ... -2.6600260e-01
1.8728569e-01 4.1819674e-01]
[ 7.8903240e-01 3.0281746e-01 1.7624519e+00 ... 4.4533429e-01
2.8690374e-01 7.4479729e-03]
...
[ 1.1210388e-44 1.2611686e-44 4.0637655e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.1210388e-44 1.2611686e-44 4.0637655e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.1210388e-44 1.2611686e-44 4.0637655e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 7.8903055e-01, 3.0284563e-01, 1.7624679e+00, ...,
1.6439064e-01, -4.7409746e-01, 1.6958097e-01],
[ 7.8903079e-01, 3.0283996e-01, 1.7624480e+00, ...,
-2.6600260e-01, 1.8728569e-01, 4.1819674e-01],
[ 7.8903240e-01, 3.0281746e-01, 1.7624519e+00, ...,
4.4533429e-01, 2.8690374e-01, 7.4479729e-03],
...,
[ 1.1210388e-44, 1.2611686e-44, 4.0637655e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.1210388e-44, 1.2611686e-44, 4.0637655e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.1210388e-44, 1.2611686e-44, 4.0637655e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'SevenNetConv_0': SevenNetConv_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[ 7.8903055e-01 3.0284563e-01 1.7624679e+00 ... 1.6439064e-01
-4.7409746e-01 1.6958097e-01]
[ 7.8903079e-01 3.0283996e-01 1.7624480e+00 ... -2.6600260e-01
1.8728569e-01 4.1819674e-01]
[ 7.8903240e-01 3.0281746e-01 1.7624519e+00 ... 4.4533429e-01
2.8690374e-01 7.4479729e-03]
...
[ 1.1210388e-44 1.2611686e-44 4.0637655e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.1210388e-44 1.2611686e-44 4.0637655e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 1.1210388e-44 1.2611686e-44 4.0637655e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 7.8903055e-01, 3.0284563e-01, 1.7624679e+00, ...,
1.6439064e-01, -4.7409746e-01, 1.6958097e-01],
[ 7.8903079e-01, 3.0283996e-01, 1.7624480e+00, ...,
-2.6600260e-01, 1.8728569e-01, 4.1819674e-01],
[ 7.8903240e-01, 3.0281746e-01, 1.7624519e+00, ...,
4.4533429e-01, 2.8690374e-01, 7.4479729e-03],
...,
[ 1.1210388e-44, 1.2611686e-44, 4.0637655e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.1210388e-44, 1.2611686e-44, 4.0637655e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 1.1210388e-44, 1.2611686e-44, 4.0637655e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'LazyInMLP_0': LazyInMLP_0input: 'x': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,,}children: 'Dense_0': Dense_0input: 'inputs': Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32):Rendering array...
,}params: 7680 parameterskernel: Array([[ 1.9611268 , 0.89812714, -1.9948477 , ..., 0.30886963,
1.1508706 , 3.4620895 ],
[ 1.8762344 , 0.6253016 , -1.980058 , ..., -0.70826226,
1.139967 , -0.7580052 ],
[-1.3364033 , 0.55748296, 0.44204178, ..., -3.708407 ,
0.45144904, 0.67026114],
...,
[ 0.10644482, 0.4301311 , 0.04070838, ..., 1.2164056 ,
-4.283859 , 3.9550807 ],
[-0.40197 , 0.56505716, 0.7024608 , ..., -0.2792783 ,
-5.6710806 , 0.0624095 ],
[-0.3675052 , 0.54707265, 0.21580404, ..., -0.57510746,
-3.7059128 , -0.69469917]], dtype=float32):Rendering array...
,}output: Array([[[ 7.35480607e-01, 7.03230560e-01, -8.26477945e-01, ...,
-1.49849844e+00, 8.28794420e-01, -1.00703466e+00],
[ 1.02315867e+00, 1.99174806e-01, -8.51171792e-01, ...,
2.99210519e-01, 7.56071627e-01, 5.14216542e-01],
[ 6.54132724e-01, -1.76375911e-01, -5.85293889e-01, ...,
9.83544767e-01, 6.02065086e-01, 4.26705986e-01],
...,
[-3.17240926e-03, -1.64806414e-02, -4.46722098e-03, ...,
-8.39055143e-03, 6.97432309e-02, 2.00373754e-02],
[-2.92137428e-03, -1.47917420e-02, -4.25948715e-03, ...,
-7.94654153e-03, 6.34378195e-02, 1.89575516e-02],
[-2.40494753e-03, -1.15049435e-02, -3.78031936e-03, ...,
-6.95143500e-03, 5.09456061e-02, 1.65958032e-02]],
[[ 7.35480607e-01, 7.03230560e-01, -8.26477945e-01, ...,
-1.49849844e+00, 8.28794420e-01, -1.00703466e+00],
[ 1.02315879e+00, 1.99175254e-01, -8.51171911e-01, ...,
2.99209505e-01, 7.56070971e-01, 5.14215589e-01],
[ 6.54131353e-01, -1.76375523e-01, -5.85292637e-01, ...,
9.83543694e-01, 6.02060676e-01, 4.26704019e-01],
...,
[-3.17240926e-03, -1.64806414e-02, -4.46722098e-03, ...,
-8.39055143e-03, 6.97432309e-02, 2.00373754e-02],
[-2.92134960e-03, -1.47916619e-02, -4.25949041e-03, ...,
-7.94653222e-03, 6.34376779e-02, 1.89574063e-02],
[-2.40497058e-03, -1.15050329e-02, -3.78032262e-03, ...,
-6.95147086e-03, 5.09458557e-02, 1.65958572e-02]],
[[ 7.35480726e-01, 7.03230560e-01, -8.26478004e-01, ...,
-1.49849820e+00, 8.28793943e-01, -1.00703466e+00],
[ 1.02315843e+00, 1.99175462e-01, -8.51171792e-01, ...,
2.99208820e-01, 7.56070375e-01, 5.14215410e-01],
[ 6.54132426e-01, -1.76375672e-01, -5.85293651e-01, ...,
9.83544528e-01, 6.02063060e-01, 4.26706076e-01],
...,
[-3.17242346e-03, -1.64807681e-02, -4.46723634e-03, ...,
-8.39057192e-03, 6.97437152e-02, 2.00373363e-02],
[-2.92137428e-03, -1.47917420e-02, -4.25948715e-03, ...,
-7.94654153e-03, 6.34378195e-02, 1.89575516e-02],
[-2.40494753e-03, -1.15049435e-02, -3.78031936e-03, ...,
-6.95143500e-03, 5.09456061e-02, 1.65958032e-02]],
...,
[[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
...,
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00]],
[[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
...,
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00]],
[[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
...,
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00]]], dtype=float32):Rendering array...
,),}output: 960x0e
[[[ 7.35480607e-01 7.03230560e-01 -8.26477945e-01 ... -1.49849844e+00
8.28794420e-01 -1.00703466e+00]
[ 1.02315867e+00 1.99174806e-01 -8.51171792e-01 ... 2.99210519e-01
7.56071627e-01 5.14216542e-01]
[ 6.54132724e-01 -1.76375911e-01 -5.85293889e-01 ... 9.83544767e-01
6.02065086e-01 4.26705986e-01]
...
[-3.17240926e-03 -1.64806414e-02 -4.46722098e-03 ... -8.39055143e-03
6.97432309e-02 2.00373754e-02]
[-2.92137428e-03 -1.47917420e-02 -4.25948715e-03 ... -7.94654153e-03
6.34378195e-02 1.89575516e-02]
[-2.40494753e-03 -1.15049435e-02 -3.78031936e-03 ... -6.95143500e-03
5.09456061e-02 1.65958032e-02]]
[[ 7.35480607e-01 7.03230560e-01 -8.26477945e-01 ... -1.49849844e+00
8.28794420e-01 -1.00703466e+00]
[ 1.02315879e+00 1.99175254e-01 -8.51171911e-01 ... 2.99209505e-01
7.56070971e-01 5.14215589e-01]
[ 6.54131353e-01 -1.76375523e-01 -5.85292637e-01 ... 9.83543694e-01
6.02060676e-01 4.26704019e-01]
...
[-3.17240926e-03 -1.64806414e-02 -4.46722098e-03 ... -8.39055143e-03
6.97432309e-02 2.00373754e-02]
[-2.92134960e-03 -1.47916619e-02 -4.25949041e-03 ... -7.94653222e-03
6.34376779e-02 1.89574063e-02]
[-2.40497058e-03 -1.15050329e-02 -3.78032262e-03 ... -6.95147086e-03
5.09458557e-02 1.65958572e-02]]
[[ 7.35480726e-01 7.03230560e-01 -8.26478004e-01 ... -1.49849820e+00
8.28793943e-01 -1.00703466e+00]
[ 1.02315843e+00 1.99175462e-01 -8.51171792e-01 ... 2.99208820e-01
7.56070375e-01 5.14215410e-01]
[ 6.54132426e-01 -1.76375672e-01 -5.85293651e-01 ... 9.83544528e-01
6.02063060e-01 4.26706076e-01]
...
[-3.17242346e-03 -1.64807681e-02 -4.46723634e-03 ... -8.39057192e-03
6.97437152e-02 2.00373363e-02]
[-2.92137428e-03 -1.47917420e-02 -4.25948715e-03 ... -7.94654153e-03
6.34378195e-02 1.89575516e-02]
[-2.40494753e-03 -1.15049435e-02 -3.78031936e-03 ... -6.95143500e-03
5.09456061e-02 1.65958032e-02]]
...
[[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
...
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]]
[[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
...
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]]
[[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
...
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]
[-3.10452795e+00 2.97421670e+00 2.47263432e+00 ... -2.52461761e-01
-3.23919907e+01 9.46277523e+00]]]key=0,): Array([[[ 7.35480607e-01, 7.03230560e-01, -8.26477945e-01, ...,
-1.49849844e+00, 8.28794420e-01, -1.00703466e+00],
[ 1.02315867e+00, 1.99174806e-01, -8.51171792e-01, ...,
2.99210519e-01, 7.56071627e-01, 5.14216542e-01],
[ 6.54132724e-01, -1.76375911e-01, -5.85293889e-01, ...,
9.83544767e-01, 6.02065086e-01, 4.26705986e-01],
...,
[-3.17240926e-03, -1.64806414e-02, -4.46722098e-03, ...,
-8.39055143e-03, 6.97432309e-02, 2.00373754e-02],
[-2.92137428e-03, -1.47917420e-02, -4.25948715e-03, ...,
-7.94654153e-03, 6.34378195e-02, 1.89575516e-02],
[-2.40494753e-03, -1.15049435e-02, -3.78031936e-03, ...,
-6.95143500e-03, 5.09456061e-02, 1.65958032e-02]],
[[ 7.35480607e-01, 7.03230560e-01, -8.26477945e-01, ...,
-1.49849844e+00, 8.28794420e-01, -1.00703466e+00],
[ 1.02315879e+00, 1.99175254e-01, -8.51171911e-01, ...,
2.99209505e-01, 7.56070971e-01, 5.14215589e-01],
[ 6.54131353e-01, -1.76375523e-01, -5.85292637e-01, ...,
9.83543694e-01, 6.02060676e-01, 4.26704019e-01],
...,
[-3.17240926e-03, -1.64806414e-02, -4.46722098e-03, ...,
-8.39055143e-03, 6.97432309e-02, 2.00373754e-02],
[-2.92134960e-03, -1.47916619e-02, -4.25949041e-03, ...,
-7.94653222e-03, 6.34376779e-02, 1.89574063e-02],
[-2.40497058e-03, -1.15050329e-02, -3.78032262e-03, ...,
-6.95147086e-03, 5.09458557e-02, 1.65958572e-02]],
[[ 7.35480726e-01, 7.03230560e-01, -8.26478004e-01, ...,
-1.49849820e+00, 8.28793943e-01, -1.00703466e+00],
[ 1.02315843e+00, 1.99175462e-01, -8.51171792e-01, ...,
2.99208820e-01, 7.56070375e-01, 5.14215410e-01],
[ 6.54132426e-01, -1.76375672e-01, -5.85293651e-01, ...,
9.83544528e-01, 6.02063060e-01, 4.26706076e-01],
...,
[-3.17242346e-03, -1.64807681e-02, -4.46723634e-03, ...,
-8.39057192e-03, 6.97437152e-02, 2.00373363e-02],
[-2.92137428e-03, -1.47917420e-02, -4.25948715e-03, ...,
-7.94654153e-03, 6.34378195e-02, 1.89575516e-02],
[-2.40494753e-03, -1.15049435e-02, -3.78031936e-03, ...,
-6.95143500e-03, 5.09456061e-02, 1.65958032e-02]],
...,
[[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
...,
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00]],
[[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
...,
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00]],
[[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
...,
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00],
[-3.10452795e+00, 2.97421670e+00, 2.47263432e+00, ...,
-2.52461761e-01, -3.23919907e+01, 9.46277523e+00]]], dtype=float32), :Rendering array...
,,),}output: 224x0e+384x1e+352x2e
[[-0.10368696 0.03684989 -0.30516163 ... 0.10380977 -0.03727242
-0.04088269]
[-0.10368773 0.03685224 -0.3051644 ... -0.06745671 -0.07696537
0.05810416]
[-0.10368752 0.03685156 -0.3051639 ... -0.01158423 0.11425027
0.02574089]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.10368696, 0.03684989, -0.30516163, ..., 0.10380977,
-0.03727242, -0.04088269],
[-0.10368773, 0.03685224, -0.3051644 , ..., -0.06745671,
-0.07696537, 0.05810416],
[-0.10368752, 0.03685156, -0.3051639 , ..., -0.01158423,
0.11425027, 0.02574089],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'linear_outro': linear_outroinput: 'weights_or_input': 224x0e+384x1e+352x2e
[[-0.10368696 0.03684989 -0.30516163 ... 0.10380977 -0.03727242
-0.04088269]
[-0.10368773 0.03685224 -0.3051644 ... -0.06745671 -0.07696537
0.05810416]
[-0.10368752 0.03685156 -0.3051639 ... -0.01158423 0.11425027
0.02574089]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.10368696, 0.03684989, -0.30516163, ..., 0.10380977,
-0.03727242, -0.04088269],
[-0.10368773, 0.03685224, -0.3051644 , ..., -0.06745671,
-0.07696537, 0.05810416],
[-0.10368752, 0.03685156, -0.3051639 , ..., -0.01158423,
0.11425027, 0.02574089],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 86016 parametersw[0,0] 224x0e,224x0e: Array([[ 2.0081546 , 1.6479783 , 1.5604763 , ..., 1.4455466 ,
-1.0169146 , -1.3716726 ],
[-2.3814116 , -0.9727504 , -1.6676872 , ..., 0.1920512 ,
-0.51205426, -2.6818907 ],
[ 1.6531345 , 1.0792857 , -0.8966044 , ..., -1.891223 ,
-1.4565222 , 1.7349514 ],
...,
[ 2.9895694 , 2.408599 , -0.29375437, ..., -1.8559114 ,
5.440316 , -5.4067397 ],
[ 1.5446353 , -0.12033591, 1.850207 , ..., -0.9702179 ,
1.4745461 , 3.9028435 ],
[-2.7847724 , -2.2506235 , -1.7330496 , ..., 5.0526257 ,
2.3648324 , 0.84812665]], dtype=float32):Rendering array...
, w[1,1] 384x1e,64x1e: Array([[-4.5358515 , 1.1497406 , 2.589901 , ..., -2.043665 ,
0.22063464, 1.3551856 ],
[-1.0841694 , -1.8803372 , 0.6187718 , ..., 3.6525974 ,
-2.4811273 , 2.0372815 ],
[-4.234338 , 2.2495852 , -3.7499642 , ..., -3.6261945 ,
2.5682158 , -0.564426 ],
...,
[ 1.5536867 , -0.52748156, -1.7037528 , ..., -1.7256256 ,
0.5423395 , 0.8801775 ],
[ 0.59128857, -0.13331643, -0.565452 , ..., 2.2760124 ,
-2.2967505 , 1.0503368 ],
[-0.9111213 , -1.1474068 , -0.18583015, ..., -0.9213156 ,
2.2737846 , 0.03138326]], dtype=float32):Rendering array...
, w[2,2] 352x2e,32x2e: Array([[-1.3098822 , -4.5079107 , -1.8133179 , ..., -0.64720494,
2.876182 , 1.0001799 ],
[-3.1235614 , 2.1366012 , -1.4442047 , ..., -1.5842416 ,
-0.0381077 , 5.034386 ],
[ 1.4682164 , -1.0672374 , 2.318073 , ..., 1.0078801 ,
-0.9195423 , 1.4017334 ],
...,
[ 2.1997006 , -1.3968203 , -1.3433715 , ..., 2.8820744 ,
-0.2825411 , 0.8039047 ],
[-1.2277429 , 0.2927603 , 4.215725 , ..., -2.0764709 ,
3.7434182 , -5.0725036 ],
[-2.206367 , -6.099845 , -1.0706434 , ..., 4.453828 ,
-0.0274033 , -0.7044329 ]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[ 0.10213755 -0.18940291 -0.35433447 ... 0.01385735 -0.20210804
-0.00687585]
[ 0.10214316 -0.18940522 -0.35430574 ... -0.15769988 0.0876667
0.09232648]
[ 0.1021881 -0.18942201 -0.35436848 ... 0.14531456 0.11435429
-0.08278326]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.10213755, -0.18940291, -0.35433447, ..., 0.01385735,
-0.20210804, -0.00687585],
[ 0.10214316, -0.18940522, -0.35430574, ..., -0.15769988,
0.0876667 , 0.09232648],
[ 0.1021881 , -0.18942201, -0.35436848, ..., 0.14531456,
0.11435429, -0.08278326],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[ 0.10213755 -0.18940291 -0.35433447 ... 0.01385735 -0.20210804
-0.00687585]
[ 0.10214316 -0.18940522 -0.35430574 ... -0.15769988 0.0876667
0.09232648]
[ 0.1021881 -0.18942201 -0.35436848 ... 0.14531456 0.11435429
-0.08278326]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.10213755, -0.18940291, -0.35433447, ..., 0.01385735,
-0.20210804, -0.00687585],
[ 0.10214316, -0.18940522, -0.35430574, ..., -0.15769988,
0.0876667 , 0.09232648],
[ 0.1021881 , -0.18942201, -0.35436848, ..., 0.14531456,
0.11435429, -0.08278326],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[-8.4898192e-01 -2.1024830e+00 -1.4092977e+00 ... 9.2710800e-02
-2.4283022e-01 -2.0878501e-02]
[-8.4897494e-01 -2.1024852e+00 -1.4092646e+00 ... -2.0720488e-01
4.2844173e-02 1.5242551e-01]
[-8.4893483e-01 -2.1024842e+00 -1.4093261e+00 ... 1.5692531e-01
1.9993284e-01 -5.7946704e-02]
...
[ 5.6051939e-45 1.4012985e-45 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 5.6051939e-45 1.4012985e-45 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 5.6051939e-45 1.4012985e-45 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-8.4898192e-01, -2.1024830e+00, -1.4092977e+00, ...,
9.2710800e-02, -2.4283022e-01, -2.0878501e-02],
[-8.4897494e-01, -2.1024852e+00, -1.4092646e+00, ...,
-2.0720488e-01, 4.2844173e-02, 1.5242551e-01],
[-8.4893483e-01, -2.1024842e+00, -1.4093261e+00, ...,
1.5692531e-01, 1.9993284e-01, -5.7946704e-02],
...,
[ 5.6051939e-45, 1.4012985e-45, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 5.6051939e-45, 1.4012985e-45, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 5.6051939e-45, 1.4012985e-45, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'self_connection': self_connectioninput: 'node_feats': 224x0e+64x1e+32x2e
[[-8.4898192e-01 -2.1024830e+00 -1.4092977e+00 ... 9.2710800e-02
-2.4283022e-01 -2.0878501e-02]
[-8.4897494e-01 -2.1024852e+00 -1.4092646e+00 ... -2.0720488e-01
4.2844173e-02 1.5242551e-01]
[-8.4893483e-01 -2.1024842e+00 -1.4093261e+00 ... 1.5692531e-01
1.9993284e-01 -5.7946704e-02]
...
[ 5.6051939e-45 1.4012985e-45 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 5.6051939e-45 1.4012985e-45 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 5.6051939e-45 1.4012985e-45 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-8.4898192e-01, -2.1024830e+00, -1.4092977e+00, ...,
9.2710800e-02, -2.4283022e-01, -2.0878501e-02],
[-8.4897494e-01, -2.1024852e+00, -1.4092646e+00, ...,
-2.0720488e-01, 4.2844173e-02, 1.5242551e-01],
[-8.4893483e-01, -2.1024842e+00, -1.4093261e+00, ...,
1.5692531e-01, 1.9993284e-01, -5.7946704e-02],
...,
[ 5.6051939e-45, 1.4012985e-45, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 5.6051939e-45, 1.4012985e-45, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 5.6051939e-45, 1.4012985e-45, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_specie': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[-4.2650878e-01 -3.8371086e-01 -4.6391952e-01 ... 1.1290850e-01
-2.9573247e-01 -2.5427029e-02]
[-4.2650732e-01 -3.8371056e-01 -4.6392095e-01 ... -2.5236219e-01
5.2181438e-02 1.8564446e-01]
[-4.2649913e-01 -3.8371062e-01 -4.6391827e-01 ... 1.9111128e-01
2.4348797e-01 -7.0570320e-02]
...
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-4.2650878e-01, -3.8371086e-01, -4.6391952e-01, ...,
1.1290850e-01, -2.9573247e-01, -2.5427029e-02],
[-4.2650732e-01, -3.8371056e-01, -4.6392095e-01, ...,
-2.5236219e-01, 5.2181438e-02, 1.8564446e-01],
[-4.2649913e-01, -3.8371062e-01, -4.6391827e-01, ...,
1.9111128e-01, 2.4348797e-01, -7.0570320e-02],
...,
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 128x0e+64x1e+32x2e
[[-4.2650878e-01 -3.8371086e-01 -4.6391952e-01 ... 1.1290850e-01
-2.9573247e-01 -2.5427029e-02]
[-4.2650732e-01 -3.8371056e-01 -4.6392095e-01 ... -2.5236219e-01
5.2181438e-02 1.8564446e-01]
[-4.2649913e-01 -3.8371062e-01 -4.6391827e-01 ... 1.9111128e-01
2.4348797e-01 -7.0570320e-02]
...
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-4.2650878e-01, -3.8371086e-01, -4.6391952e-01, ...,
1.1290850e-01, -2.9573247e-01, -2.5427029e-02],
[-4.2650732e-01, -3.8371056e-01, -4.6392095e-01, ...,
-2.5236219e-01, 5.2181438e-02, 1.8564446e-01],
[-4.2649913e-01, -3.8371062e-01, -4.6391827e-01, ...,
1.9111128e-01, 2.4348797e-01, -7.0570320e-02],
...,
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, None,),), 'layer_3': layer_3input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-4.2650878e-01 -3.8371086e-01 -4.6391952e-01 ... 1.1290850e-01
-2.9573247e-01 -2.5427029e-02]
[-4.2650732e-01 -3.8371056e-01 -4.6392095e-01 ... -2.5236219e-01
5.2181438e-02 1.8564446e-01]
[-4.2649913e-01 -3.8371062e-01 -4.6391827e-01 ... 1.9111128e-01
2.4348797e-01 -7.0570320e-02]
...
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-4.2650878e-01, -3.8371086e-01, -4.6391952e-01, ...,
1.1290850e-01, -2.9573247e-01, -2.5427029e-02],
[-4.2650732e-01, -3.8371056e-01, -4.6392095e-01, ...,
-2.5236219e-01, 5.2181438e-02, 1.8564446e-01],
[-4.2649913e-01, -3.8371062e-01, -4.6391827e-01, ...,
1.9111128e-01, 2.4348797e-01, -7.0570320e-02],
...,
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_species': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'interaction': interactioninput: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-4.2650878e-01 -3.8371086e-01 -4.6391952e-01 ... 1.1290850e-01
-2.9573247e-01 -2.5427029e-02]
[-4.2650732e-01 -3.8371056e-01 -4.6392095e-01 ... -2.5236219e-01
5.2181438e-02 1.8564446e-01]
[-4.2649913e-01 -3.8371062e-01 -4.6391827e-01 ... 1.9111128e-01
2.4348797e-01 -7.0570320e-02]
...
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-4.2650878e-01, -3.8371086e-01, -4.6391952e-01, ...,
1.1290850e-01, -2.9573247e-01, -2.5427029e-02],
[-4.2650732e-01, -3.8371056e-01, -4.6392095e-01, ...,
-2.5236219e-01, 5.2181438e-02, 1.8564446e-01],
[-4.2649913e-01, -3.8371062e-01, -4.6391827e-01, ...,
1.9111128e-01, 2.4348797e-01, -7.0570320e-02],
...,
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'resid_adapter': resid_adapterinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[-4.2650878e-01 -3.8371086e-01 -4.6391952e-01 ... 1.1290850e-01
-2.9573247e-01 -2.5427029e-02]
[-4.2650732e-01 -3.8371056e-01 -4.6392095e-01 ... -2.5236219e-01
5.2181438e-02 1.8564446e-01]
[-4.2649913e-01 -3.8371062e-01 -4.6391827e-01 ... 1.9111128e-01
2.4348797e-01 -7.0570320e-02]
...
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-4.2650878e-01, -3.8371086e-01, -4.6391952e-01, ...,
1.1290850e-01, -2.9573247e-01, -2.5427029e-02],
[-4.2650732e-01, -3.8371056e-01, -4.6392095e-01, ...,
-2.5236219e-01, 5.2181438e-02, 1.8564446e-01],
[-4.2649913e-01, -3.8371062e-01, -4.6391827e-01, ...,
1.9111128e-01, 2.4348797e-01, -7.0570320e-02],
...,
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 33792 parametersw[0,0] 128x0e,224x0e: Array([[-0.12081312, 0.21433187, 1.4421564 , ..., 0.48700368,
0.83675104, 2.5179884 ],
[-0.00837716, -0.60929865, 3.6444187 , ..., -1.095506 ,
-2.9168863 , 0.76664406],
[ 1.6383092 , -3.41747 , 0.1877312 , ..., -2.2291133 ,
-0.70872074, -0.67117375],
...,
[ 0.22261605, -1.6185126 , -0.29067388, ..., -0.30721062,
-0.936053 , -6.319776 ],
[ 0.96031153, 0.3580432 , -2.3184853 , ..., -3.5137513 ,
0.7674257 , 1.8014185 ],
[-1.5907912 , -1.7231196 , 0.751697 , ..., -0.34286314,
-3.2077777 , -3.3623428 ]], dtype=float32):Rendering array...
, w[1,1] 64x1e,64x1e: Array([[-2.7112663 , -0.9643581 , 2.9698734 , ..., -1.6670817 ,
-0.6212921 , 0.7902799 ],
[ 1.22522 , -2.1814368 , 0.55360013, ..., -0.83167875,
-0.0256775 , 1.8732722 ],
[ 0.49691385, 0.24650456, 1.1760592 , ..., 0.6746525 ,
1.6669806 , 2.3433416 ],
...,
[-4.331458 , 5.7505627 , 2.1066272 , ..., 1.3505996 ,
-2.1723619 , 2.1864228 ],
[-2.0217304 , -3.1857903 , 2.0970798 , ..., -1.8917772 ,
-1.419903 , -2.8695233 ],
[ 0.63794476, 1.5040748 , -0.4937292 , ..., 1.2936034 ,
3.5097804 , 2.3465707 ]], dtype=float32):Rendering array...
, w[2,2] 32x2e,32x2e: Array([[ 0.23311885, 2.5184479 , 0.35841894, ..., 1.2846606 ,
0.6560974 , -0.598776 ],
[ 0.7402837 , 1.833609 , 2.4995692 , ..., -0.63918394,
-2.8790867 , 1.724812 ],
[ 1.0263193 , -2.7364955 , 0.9500314 , ..., -0.22096811,
1.1111536 , 0.79498106],
...,
[ 1.2500571 , 3.3613405 , 0.46471608, ..., 1.5784024 ,
3.1365042 , -2.0092516 ],
[-0.22173691, 0.08692293, -0.47295108, ..., 0.3666285 ,
-3.3435378 , 0.50542533],
[ 0.2489541 , 0.22905599, -0.8613533 , ..., -0.14420761,
2.0106645 , 0.28263944]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[ 1.61326423e-01 -1.10758436e+00 -2.88294345e-01 ... 3.05590749e-01
-2.01408789e-01 1.25153810e-01]
[ 1.61348253e-01 -1.10753381e+00 -2.88203180e-01 ... -1.07978925e-01
-7.42529705e-02 3.63931805e-01]
[ 1.61315918e-01 -1.10758400e+00 -2.88226455e-01 ... 1.94161430e-01
2.75622606e-01 1.89478144e-01]
...
[-2.80259693e-45 -7.00649232e-45 -3.08285662e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -7.00649232e-45 -3.08285662e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]
[-2.80259693e-45 -7.00649232e-45 -3.08285662e-44 ... 0.00000000e+00
0.00000000e+00 0.00000000e+00]]key=0,): Array([[ 1.61326423e-01, -1.10758436e+00, -2.88294345e-01, ...,
3.05590749e-01, -2.01408789e-01, 1.25153810e-01],
[ 1.61348253e-01, -1.10753381e+00, -2.88203180e-01, ...,
-1.07978925e-01, -7.42529705e-02, 3.63931805e-01],
[ 1.61315918e-01, -1.10758400e+00, -2.88226455e-01, ...,
1.94161430e-01, 2.75622606e-01, 1.89478144e-01],
...,
[-2.80259693e-45, -7.00649232e-45, -3.08285662e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -7.00649232e-45, -3.08285662e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00],
[-2.80259693e-45, -7.00649232e-45, -3.08285662e-44, ...,
0.00000000e+00, 0.00000000e+00, 0.00000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'SimpleInteraction_0': SimpleInteraction_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-4.2650878e-01 -3.8371086e-01 -4.6391952e-01 ... 1.1290850e-01
-2.9573247e-01 -2.5427029e-02]
[-4.2650732e-01 -3.8371056e-01 -4.6392095e-01 ... -2.5236219e-01
5.2181438e-02 1.8564446e-01]
[-4.2649913e-01 -3.8371062e-01 -4.6391827e-01 ... 1.9111128e-01
2.4348797e-01 -7.0570320e-02]
...
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-4.2650878e-01, -3.8371086e-01, -4.6391952e-01, ...,
1.1290850e-01, -2.9573247e-01, -2.5427029e-02],
[-4.2650732e-01, -3.8371056e-01, -4.6392095e-01, ...,
-2.5236219e-01, 5.2181438e-02, 1.8564446e-01],
[-4.2649913e-01, -3.8371062e-01, -4.6391827e-01, ...,
1.9111128e-01, 2.4348797e-01, -7.0570320e-02],
...,
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'linear_intro': linear_introinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[-4.2650878e-01 -3.8371086e-01 -4.6391952e-01 ... 1.1290850e-01
-2.9573247e-01 -2.5427029e-02]
[-4.2650732e-01 -3.8371056e-01 -4.6392095e-01 ... -2.5236219e-01
5.2181438e-02 1.8564446e-01]
[-4.2649913e-01 -3.8371062e-01 -4.6391827e-01 ... 1.9111128e-01
2.4348797e-01 -7.0570320e-02]
...
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[ 4.2038954e-45 0.0000000e+00 1.4012985e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-4.2650878e-01, -3.8371086e-01, -4.6391952e-01, ...,
1.1290850e-01, -2.9573247e-01, -2.5427029e-02],
[-4.2650732e-01, -3.8371056e-01, -4.6392095e-01, ...,
-2.5236219e-01, 5.2181438e-02, 1.8564446e-01],
[-4.2649913e-01, -3.8371062e-01, -4.6391827e-01, ...,
1.9111128e-01, 2.4348797e-01, -7.0570320e-02],
...,
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 4.2038954e-45, 0.0000000e+00, 1.4012985e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 21504 parametersw[0,0] 128x0e,128x0e: Array([[ 1.140101 , 2.0222821 , -3.1304958 , ..., 1.1466542 ,
1.7076294 , -0.21663147],
[-3.198693 , 2.8868864 , -0.3604192 , ..., -0.7823056 ,
0.6240356 , 0.98817015],
[-4.6553483 , 0.6639529 , -1.5882826 , ..., -0.1580747 ,
0.34259212, -0.9970385 ],
...,
[-0.36834115, 2.3281004 , -0.15078871, ..., -0.99443746,
2.3609061 , -2.7193348 ],
[-2.0036862 , -2.430595 , -0.5361116 , ..., -0.35383248,
-0.13564959, -1.6325415 ],
[-5.8481436 , 2.207725 , 0.9011133 , ..., -0.6096626 ,
-0.51655275, -0.6637965 ]], dtype=float32):Rendering array...
, w[1,1] 64x1e,64x1e: Array([[-6.1574583 , 0.5306572 , -6.2926936 , ..., -0.02864111,
-0.40215862, 5.589474 ],
[-3.3362794 , -6.298284 , -3.7303557 , ..., 2.2301488 ,
1.0625205 , 1.2262591 ],
[-2.5761666 , 1.2296215 , -0.37371475, ..., 3.42179 ,
-0.6136429 , 5.521514 ],
...,
[-2.8534966 , 0.6100542 , 2.76881 , ..., -1.0152637 ,
-4.398335 , 3.590753 ],
[-0.06075297, 1.8847165 , 4.0440626 , ..., -0.10650878,
0.69415295, -2.9796302 ],
[ 0.44205856, 3.0723534 , 1.9417149 , ..., -3.4233217 ,
2.7222688 , 0.73819387]], dtype=float32):Rendering array...
, w[2,2] 32x2e,32x2e: Array([[-3.9599621 , -2.1413295 , 5.519057 , ..., 0.873332 ,
3.4918246 , 3.0262551 ],
[ 1.1177975 , 2.9307213 , -3.712386 , ..., -5.507332 ,
-2.4439375 , 3.9713829 ],
[ 0.7814051 , 2.3921692 , 0.28386188, ..., 1.8599669 ,
-0.90636504, -1.0435562 ],
...,
[ 0.94891405, 0.05201203, 1.257389 , ..., 3.6583638 ,
1.3928108 , -0.59312904],
[-2.7363954 , 0.7783992 , -0.04990867, ..., 1.9942182 ,
4.239317 , -3.9982047 ],
[-1.4600741 , -1.0381477 , 2.0245078 , ..., 2.7030635 ,
1.8954883 , -0.8860114 ]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[-7.6021664e-02 9.0243387e-01 -2.5832915e-01 ... -6.4266153e-02
3.2077557e-01 -2.6308647e-01]
[-7.6075017e-02 9.0241635e-01 -2.5825357e-01 ... 7.7688172e-02
-2.2599772e-01 -3.4517002e-01]
[-7.6007962e-02 9.0242106e-01 -2.5826445e-01 ... -4.0349975e-01
-9.4813772e-02 -6.7549229e-02]
...
[-5.3249342e-44 6.3058431e-44 1.2611686e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.3249342e-44 6.3058431e-44 1.2611686e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.3249342e-44 6.3058431e-44 1.2611686e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-7.6021664e-02, 9.0243387e-01, -2.5832915e-01, ...,
-6.4266153e-02, 3.2077557e-01, -2.6308647e-01],
[-7.6075017e-02, 9.0241635e-01, -2.5825357e-01, ...,
7.7688172e-02, -2.2599772e-01, -3.4517002e-01],
[-7.6007962e-02, 9.0242106e-01, -2.5826445e-01, ...,
-4.0349975e-01, -9.4813772e-02, -6.7549229e-02],
...,
[-5.3249342e-44, 6.3058431e-44, 1.2611686e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.3249342e-44, 6.3058431e-44, 1.2611686e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.3249342e-44, 6.3058431e-44, 1.2611686e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'SevenNetConv_0': SevenNetConv_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-7.6021664e-02 9.0243387e-01 -2.5832915e-01 ... -6.4266153e-02
3.2077557e-01 -2.6308647e-01]
[-7.6075017e-02 9.0241635e-01 -2.5825357e-01 ... 7.7688172e-02
-2.2599772e-01 -3.4517002e-01]
[-7.6007962e-02 9.0242106e-01 -2.5826445e-01 ... -4.0349975e-01
-9.4813772e-02 -6.7549229e-02]
...
[-5.3249342e-44 6.3058431e-44 1.2611686e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.3249342e-44 6.3058431e-44 1.2611686e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-5.3249342e-44 6.3058431e-44 1.2611686e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-7.6021664e-02, 9.0243387e-01, -2.5832915e-01, ...,
-6.4266153e-02, 3.2077557e-01, -2.6308647e-01],
[-7.6075017e-02, 9.0241635e-01, -2.5825357e-01, ...,
7.7688172e-02, -2.2599772e-01, -3.4517002e-01],
[-7.6007962e-02, 9.0242106e-01, -2.5826445e-01, ...,
-4.0349975e-01, -9.4813772e-02, -6.7549229e-02],
...,
[-5.3249342e-44, 6.3058431e-44, 1.2611686e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.3249342e-44, 6.3058431e-44, 1.2611686e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-5.3249342e-44, 6.3058431e-44, 1.2611686e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'LazyInMLP_0': LazyInMLP_0input: 'x': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,,}children: 'Dense_0': Dense_0input: 'inputs': Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32):Rendering array...
,}params: 7680 parameterskernel: Array([[ 1.3343583 , -0.02900918, 0.982258 , ..., -3.8649988 ,
1.6852164 , -2.8866332 ],
[ 3.663301 , -2.4428883 , 2.2433438 , ..., -3.0956182 ,
0.4433875 , -7.820587 ],
[ 4.800418 , -1.3318237 , -0.17254679, ..., 1.5635651 ,
0.811215 , -9.068334 ],
...,
[-0.16307653, -0.05829594, 4.855029 , ..., -3.9734268 ,
-7.034069 , -5.6165075 ],
[ 0.8679783 , -0.23416385, 4.688869 , ..., 1.4813001 ,
-0.89909977, -5.801539 ],
[ 1.2308747 , -0.53154147, 1.1538248 , ..., 2.3267426 ,
0.6285008 , -0.76400554]], dtype=float32):Rendering array...
,}output: Array([[[ 2.0049131e+00, -1.1143405e+00, -1.4625034e-01, ...,
2.1875381e+00, 3.9032393e+00, -1.8773681e+00],
[ 1.6202301e-01, -4.9090710e-01, 1.1302884e+00, ...,
-1.1169595e+00, 2.4838421e-01, -1.9697804e+00],
[-4.7395071e-01, 2.9573953e-02, 1.5118444e-01, ...,
-2.0817914e+00, -7.4348491e-01, 3.5938230e-01],
...,
[ 6.4781094e-03, 1.3792948e-02, -3.2066371e-02, ...,
-7.3198252e-03, 6.5112464e-02, -7.4816644e-02],
[ 6.3839490e-03, 1.2510829e-02, -2.9817669e-02, ...,
-6.3863890e-03, 5.9683163e-02, -6.7730650e-02],
[ 6.0083596e-03, 9.9889059e-03, -2.5123840e-02, ...,
-4.6563423e-03, 4.8724260e-02, -5.3767070e-02]],
[[ 2.0049131e+00, -1.1143405e+00, -1.4625034e-01, ...,
2.1875381e+00, 3.9032393e+00, -1.8773681e+00],
[ 1.6202399e-01, -4.9090755e-01, 1.1302884e+00, ...,
-1.1169568e+00, 2.4838677e-01, -1.9697807e+00],
[-4.7394973e-01, 2.9574307e-02, 1.5118416e-01, ...,
-2.0817871e+00, -7.4348247e-01, 3.5938433e-01],
...,
[ 6.4781094e-03, 1.3792948e-02, -3.2066371e-02, ...,
-7.3198252e-03, 6.5112464e-02, -7.4816644e-02],
[ 6.3838963e-03, 1.2510774e-02, -2.9817784e-02, ...,
-6.3862670e-03, 5.9683014e-02, -6.7729861e-02],
[ 6.0084052e-03, 9.9889673e-03, -2.5123879e-02, ...,
-4.6563623e-03, 4.8724581e-02, -5.3767640e-02]],
[[ 2.0049126e+00, -1.1143405e+00, -1.4624956e-01, ...,
2.1875381e+00, 3.9032390e+00, -1.8773682e+00],
[ 1.6202453e-01, -4.9090767e-01, 1.1302887e+00, ...,
-1.1169556e+00, 2.4838746e-01, -1.9697816e+00],
[-4.7395027e-01, 2.9573921e-02, 1.5118535e-01, ...,
-2.0817907e+00, -7.4348515e-01, 3.5938153e-01],
...,
[ 6.4780959e-03, 1.3793043e-02, -3.2066599e-02, ...,
-7.3198299e-03, 6.5112941e-02, -7.4816957e-02],
[ 6.3839490e-03, 1.2510829e-02, -2.9817669e-02, ...,
-6.3863890e-03, 5.9683163e-02, -6.7730650e-02],
[ 6.0083596e-03, 9.9889059e-03, -2.5123840e-02, ...,
-4.6563423e-03, 4.8724260e-02, -5.3767070e-02]],
...,
[[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
...,
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01]],
[[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
...,
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01]],
[[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
...,
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01]]], dtype=float32):Rendering array...
,),}output: 960x0e
[[[ 2.0049131e+00 -1.1143405e+00 -1.4625034e-01 ... 2.1875381e+00
3.9032393e+00 -1.8773681e+00]
[ 1.6202301e-01 -4.9090710e-01 1.1302884e+00 ... -1.1169595e+00
2.4838421e-01 -1.9697804e+00]
[-4.7395071e-01 2.9573953e-02 1.5118444e-01 ... -2.0817914e+00
-7.4348491e-01 3.5938230e-01]
...
[ 6.4781094e-03 1.3792948e-02 -3.2066371e-02 ... -7.3198252e-03
6.5112464e-02 -7.4816644e-02]
[ 6.3839490e-03 1.2510829e-02 -2.9817669e-02 ... -6.3863890e-03
5.9683163e-02 -6.7730650e-02]
[ 6.0083596e-03 9.9889059e-03 -2.5123840e-02 ... -4.6563423e-03
4.8724260e-02 -5.3767070e-02]]
[[ 2.0049131e+00 -1.1143405e+00 -1.4625034e-01 ... 2.1875381e+00
3.9032393e+00 -1.8773681e+00]
[ 1.6202399e-01 -4.9090755e-01 1.1302884e+00 ... -1.1169568e+00
2.4838677e-01 -1.9697807e+00]
[-4.7394973e-01 2.9574307e-02 1.5118416e-01 ... -2.0817871e+00
-7.4348247e-01 3.5938433e-01]
...
[ 6.4781094e-03 1.3792948e-02 -3.2066371e-02 ... -7.3198252e-03
6.5112464e-02 -7.4816644e-02]
[ 6.3838963e-03 1.2510774e-02 -2.9817784e-02 ... -6.3862670e-03
5.9683014e-02 -6.7729861e-02]
[ 6.0084052e-03 9.9889673e-03 -2.5123879e-02 ... -4.6563623e-03
4.8724581e-02 -5.3767640e-02]]
[[ 2.0049126e+00 -1.1143405e+00 -1.4624956e-01 ... 2.1875381e+00
3.9032390e+00 -1.8773682e+00]
[ 1.6202453e-01 -4.9090767e-01 1.1302887e+00 ... -1.1169556e+00
2.4838746e-01 -1.9697816e+00]
[-4.7395027e-01 2.9573921e-02 1.5118535e-01 ... -2.0817907e+00
-7.4348515e-01 3.5938153e-01]
...
[ 6.4780959e-03 1.3793043e-02 -3.2066599e-02 ... -7.3198299e-03
6.5112941e-02 -7.4816957e-02]
[ 6.3839490e-03 1.2510829e-02 -2.9817669e-02 ... -6.3863890e-03
5.9683163e-02 -6.7730650e-02]
[ 6.0083596e-03 9.9889059e-03 -2.5123840e-02 ... -4.6563423e-03
4.8724260e-02 -5.3767070e-02]]
...
[[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
...
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]]
[[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
...
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]]
[[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
...
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]
[ 1.0398389e+01 -2.6180766e+00 2.5441547e+01 ... -9.5155420e+00
-2.3202076e+01 -4.0813438e+01]]]key=0,): Array([[[ 2.0049131e+00, -1.1143405e+00, -1.4625034e-01, ...,
2.1875381e+00, 3.9032393e+00, -1.8773681e+00],
[ 1.6202301e-01, -4.9090710e-01, 1.1302884e+00, ...,
-1.1169595e+00, 2.4838421e-01, -1.9697804e+00],
[-4.7395071e-01, 2.9573953e-02, 1.5118444e-01, ...,
-2.0817914e+00, -7.4348491e-01, 3.5938230e-01],
...,
[ 6.4781094e-03, 1.3792948e-02, -3.2066371e-02, ...,
-7.3198252e-03, 6.5112464e-02, -7.4816644e-02],
[ 6.3839490e-03, 1.2510829e-02, -2.9817669e-02, ...,
-6.3863890e-03, 5.9683163e-02, -6.7730650e-02],
[ 6.0083596e-03, 9.9889059e-03, -2.5123840e-02, ...,
-4.6563423e-03, 4.8724260e-02, -5.3767070e-02]],
[[ 2.0049131e+00, -1.1143405e+00, -1.4625034e-01, ...,
2.1875381e+00, 3.9032393e+00, -1.8773681e+00],
[ 1.6202399e-01, -4.9090755e-01, 1.1302884e+00, ...,
-1.1169568e+00, 2.4838677e-01, -1.9697807e+00],
[-4.7394973e-01, 2.9574307e-02, 1.5118416e-01, ...,
-2.0817871e+00, -7.4348247e-01, 3.5938433e-01],
...,
[ 6.4781094e-03, 1.3792948e-02, -3.2066371e-02, ...,
-7.3198252e-03, 6.5112464e-02, -7.4816644e-02],
[ 6.3838963e-03, 1.2510774e-02, -2.9817784e-02, ...,
-6.3862670e-03, 5.9683014e-02, -6.7729861e-02],
[ 6.0084052e-03, 9.9889673e-03, -2.5123879e-02, ...,
-4.6563623e-03, 4.8724581e-02, -5.3767640e-02]],
[[ 2.0049126e+00, -1.1143405e+00, -1.4624956e-01, ...,
2.1875381e+00, 3.9032390e+00, -1.8773682e+00],
[ 1.6202453e-01, -4.9090767e-01, 1.1302887e+00, ...,
-1.1169556e+00, 2.4838746e-01, -1.9697816e+00],
[-4.7395027e-01, 2.9573921e-02, 1.5118535e-01, ...,
-2.0817907e+00, -7.4348515e-01, 3.5938153e-01],
...,
[ 6.4780959e-03, 1.3793043e-02, -3.2066599e-02, ...,
-7.3198299e-03, 6.5112941e-02, -7.4816957e-02],
[ 6.3839490e-03, 1.2510829e-02, -2.9817669e-02, ...,
-6.3863890e-03, 5.9683163e-02, -6.7730650e-02],
[ 6.0083596e-03, 9.9889059e-03, -2.5123840e-02, ...,
-4.6563423e-03, 4.8724260e-02, -5.3767070e-02]],
...,
[[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
...,
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01]],
[[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
...,
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01]],
[[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
...,
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01],
[ 1.0398389e+01, -2.6180766e+00, 2.5441547e+01, ...,
-9.5155420e+00, -2.3202076e+01, -4.0813438e+01]]], dtype=float32), :Rendering array...
,,),}output: 224x0e+384x1e+352x2e
[[ 0.32743195 0.29161724 0.14751874 ... 0.04700954 0.16613697
-0.00654583]
[ 0.32742074 0.29162478 0.14753897 ... 0.11454242 -0.12122759
-0.04544165]
[ 0.3274256 0.29163674 0.14754343 ... -0.13466321 -0.04487401
0.09833551]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.32743195, 0.29161724, 0.14751874, ..., 0.04700954,
0.16613697, -0.00654583],
[ 0.32742074, 0.29162478, 0.14753897, ..., 0.11454242,
-0.12122759, -0.04544165],
[ 0.3274256 , 0.29163674, 0.14754343, ..., -0.13466321,
-0.04487401, 0.09833551],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'linear_outro': linear_outroinput: 'weights_or_input': 224x0e+384x1e+352x2e
[[ 0.32743195 0.29161724 0.14751874 ... 0.04700954 0.16613697
-0.00654583]
[ 0.32742074 0.29162478 0.14753897 ... 0.11454242 -0.12122759
-0.04544165]
[ 0.3274256 0.29163674 0.14754343 ... -0.13466321 -0.04487401
0.09833551]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.32743195, 0.29161724, 0.14751874, ..., 0.04700954,
0.16613697, -0.00654583],
[ 0.32742074, 0.29162478, 0.14753897, ..., 0.11454242,
-0.12122759, -0.04544165],
[ 0.3274256 , 0.29163674, 0.14754343, ..., -0.13466321,
-0.04487401, 0.09833551],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 86016 parametersw[0,0] 224x0e,224x0e: Array([[ 1.9669358 , 2.9423544 , 0.19496085, ..., 0.73976904,
1.6372558 , -1.2503364 ],
[-0.6908223 , 0.7870249 , 1.1638534 , ..., -0.4025112 ,
-4.771617 , 1.1944067 ],
[-0.46339673, -2.2016704 , 0.06319727, ..., 2.756306 ,
0.15175773, 5.9208183 ],
...,
[-0.37803346, 4.1615434 , -1.525748 , ..., 2.7360542 ,
0.89899254, -2.888936 ],
[ 0.03935861, 2.492835 , -0.11176654, ..., -0.69757986,
1.6231072 , -1.5887352 ],
[-0.8326037 , -2.4217205 , -2.889763 , ..., 0.8011354 ,
-1.376752 , -0.8938622 ]], dtype=float32):Rendering array...
, w[1,1] 384x1e,64x1e: Array([[ 2.6754918 , 0.6090251 , -1.9812808 , ..., -4.481344 ,
-1.8871554 , 3.6247041 ],
[-1.1990739 , -0.38316777, 4.051298 , ..., -1.3397157 ,
-0.9165211 , -1.2980297 ],
[-0.05759763, 1.2653302 , -1.5408715 , ..., -0.81410015,
2.3423615 , 4.3030977 ],
...,
[-0.85361755, -0.14409246, 1.0777223 , ..., 1.1357626 ,
0.44438794, 2.4271173 ],
[ 0.24053141, -0.3990412 , 1.3709172 , ..., 0.8270215 ,
-0.07329489, 1.295343 ],
[-0.6063715 , -0.79813457, -0.47160372, ..., -0.5156604 ,
-0.51939523, -1.8995527 ]], dtype=float32):Rendering array...
, w[2,2] 352x2e,32x2e: Array([[ 6.3622594 , -1.29415 , 1.2047727 , ..., 4.7288013 ,
-0.88141096, 0.18366015],
[-2.0154412 , -4.84606 , 6.55945 , ..., -4.882003 ,
2.7193513 , 6.974478 ],
[-0.93180823, 4.2919602 , 1.4147624 , ..., -0.3700378 ,
2.1703444 , 0.28357494],
...,
[-2.133009 , 4.717238 , 0.60007834, ..., 3.8875234 ,
3.0682204 , -2.3200848 ],
[-2.0814712 , 2.6006532 , -2.6173725 , ..., -5.443099 ,
2.998717 , 0.1135042 ],
[-1.8384115 , 0.4790442 , 3.2373917 , ..., -6.6073313 ,
3.9630156 , -1.5991372 ]], dtype=float32):Rendering array...
,}output: 224x0e+64x1e+32x2e
[[-0.07027932 -1.0413275 -0.79156226 ... -1.1177028 -0.29591656
0.00784441]
[-0.07025135 -1.0414474 -0.79140705 ... -0.07727201 0.98962295
-0.59266716]
[-0.07035349 -1.0415542 -0.79146016 ... 0.36700404 -0.69325143
-0.8492176 ]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.07027932, -1.0413275 , -0.79156226, ..., -1.1177028 ,
-0.29591656, 0.00784441],
[-0.07025135, -1.0414474 , -0.79140705, ..., -0.07727201,
0.98962295, -0.59266716],
[-0.07035349, -1.0415542 , -0.79146016, ..., 0.36700404,
-0.69325143, -0.8492176 ],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[-0.07027932 -1.0413275 -0.79156226 ... -1.1177028 -0.29591656
0.00784441]
[-0.07025135 -1.0414474 -0.79140705 ... -0.07727201 0.98962295
-0.59266716]
[-0.07035349 -1.0415542 -0.79146016 ... 0.36700404 -0.69325143
-0.8492176 ]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.07027932, -1.0413275 , -0.79156226, ..., -1.1177028 ,
-0.29591656, 0.00784441],
[-0.07025135, -1.0414474 , -0.79140705, ..., -0.07727201,
0.98962295, -0.59266716],
[-0.07035349, -1.0415542 , -0.79146016, ..., 0.36700404,
-0.69325143, -0.8492176 ],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 224x0e+64x1e+32x2e
[[ 9.1047101e-02 -2.1489120e+00 -1.0798566e+00 ... -8.1211209e-01
-4.9732536e-01 1.3299821e-01]
[ 9.1096908e-02 -2.1489811e+00 -1.0796102e+00 ... -1.8525094e-01
9.1536999e-01 -2.2873536e-01]
[ 9.0962425e-02 -2.1491382e+00 -1.0796866e+00 ... 5.6116545e-01
-4.1762882e-01 -6.5973943e-01]
...
[-2.8025969e-45 -7.0064923e-45 -3.0828566e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -7.0064923e-45 -3.0828566e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -7.0064923e-45 -3.0828566e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 9.1047101e-02, -2.1489120e+00, -1.0798566e+00, ...,
-8.1211209e-01, -4.9732536e-01, 1.3299821e-01],
[ 9.1096908e-02, -2.1489811e+00, -1.0796102e+00, ...,
-1.8525094e-01, 9.1536999e-01, -2.2873536e-01],
[ 9.0962425e-02, -2.1491382e+00, -1.0796866e+00, ...,
5.6116545e-01, -4.1762882e-01, -6.5973943e-01],
...,
[-2.8025969e-45, -7.0064923e-45, -3.0828566e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -7.0064923e-45, -3.0828566e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -7.0064923e-45, -3.0828566e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'self_connection': self_connectioninput: 'node_feats': 224x0e+64x1e+32x2e
[[ 9.1047101e-02 -2.1489120e+00 -1.0798566e+00 ... -8.1211209e-01
-4.9732536e-01 1.3299821e-01]
[ 9.1096908e-02 -2.1489811e+00 -1.0796102e+00 ... -1.8525094e-01
9.1536999e-01 -2.2873536e-01]
[ 9.0962425e-02 -2.1491382e+00 -1.0796866e+00 ... 5.6116545e-01
-4.1762882e-01 -6.5973943e-01]
...
[-2.8025969e-45 -7.0064923e-45 -3.0828566e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -7.0064923e-45 -3.0828566e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -7.0064923e-45 -3.0828566e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 9.1047101e-02, -2.1489120e+00, -1.0798566e+00, ...,
-8.1211209e-01, -4.9732536e-01, 1.3299821e-01],
[ 9.1096908e-02, -2.1489811e+00, -1.0796102e+00, ...,
-1.8525094e-01, 9.1536999e-01, -2.2873536e-01],
[ 9.0962425e-02, -2.1491382e+00, -1.0796866e+00, ...,
5.6116545e-01, -4.1762882e-01, -6.5973943e-01],
...,
[-2.8025969e-45, -7.0064923e-45, -3.0828566e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -7.0064923e-45, -3.0828566e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -7.0064923e-45, -3.0828566e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_specie': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[ 7.9795286e-02 -3.7624994e-01 -4.5900899e-01 ... 2.6527405e-01
1.6244988e-01 -4.3443482e-02]
[ 7.9840839e-02 -3.7623879e-01 -4.5898870e-01 ... 6.0503438e-02
-2.9896224e-01 7.4705563e-02]
[ 7.9717852e-02 -3.7621337e-01 -4.5899495e-01 ... -1.8330802e-01
1.3642092e-01 2.1550779e-01]
...
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 7.9795286e-02, -3.7624994e-01, -4.5900899e-01, ...,
2.6527405e-01, 1.6244988e-01, -4.3443482e-02],
[ 7.9840839e-02, -3.7623879e-01, -4.5898870e-01, ...,
6.0503438e-02, -2.9896224e-01, 7.4705563e-02],
[ 7.9717852e-02, -3.7621337e-01, -4.5899495e-01, ...,
-1.8330802e-01, 1.3642092e-01, 2.1550779e-01],
...,
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,),}output: 128x0e+64x1e+32x2e
[[ 7.9795286e-02 -3.7624994e-01 -4.5900899e-01 ... 2.6527405e-01
1.6244988e-01 -4.3443482e-02]
[ 7.9840839e-02 -3.7623879e-01 -4.5898870e-01 ... 6.0503438e-02
-2.9896224e-01 7.4705563e-02]
[ 7.9717852e-02 -3.7621337e-01 -4.5899495e-01 ... -1.8330802e-01
1.3642092e-01 2.1550779e-01]
...
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 7.9795286e-02, -3.7624994e-01, -4.5900899e-01, ...,
2.6527405e-01, 1.6244988e-01, -4.3443482e-02],
[ 7.9840839e-02, -3.7623879e-01, -4.5898870e-01, ...,
6.0503438e-02, -2.9896224e-01, 7.4705563e-02],
[ 7.9717852e-02, -3.7621337e-01, -4.5899495e-01, ...,
-1.8330802e-01, 1.3642092e-01, 2.1550779e-01],
...,
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, None,),), 'layer_4': layer_4input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[ 7.9795286e-02 -3.7624994e-01 -4.5900899e-01 ... 2.6527405e-01
1.6244988e-01 -4.3443482e-02]
[ 7.9840839e-02 -3.7623879e-01 -4.5898870e-01 ... 6.0503438e-02
-2.9896224e-01 7.4705563e-02]
[ 7.9717852e-02 -3.7621337e-01 -4.5899495e-01 ... -1.8330802e-01
1.3642092e-01 2.1550779e-01]
...
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 7.9795286e-02, -3.7624994e-01, -4.5900899e-01, ...,
2.6527405e-01, 1.6244988e-01, -4.3443482e-02],
[ 7.9840839e-02, -3.7623879e-01, -4.5898870e-01, ...,
6.0503438e-02, -2.9896224e-01, 7.4705563e-02],
[ 7.9717852e-02, -3.7621337e-01, -4.5899495e-01, ...,
-1.8330802e-01, 1.3642092e-01, 2.1550779e-01],
...,
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_species': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'interaction': interactioninput: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[ 7.9795286e-02 -3.7624994e-01 -4.5900899e-01 ... 2.6527405e-01
1.6244988e-01 -4.3443482e-02]
[ 7.9840839e-02 -3.7623879e-01 -4.5898870e-01 ... 6.0503438e-02
-2.9896224e-01 7.4705563e-02]
[ 7.9717852e-02 -3.7621337e-01 -4.5899495e-01 ... -1.8330802e-01
1.3642092e-01 2.1550779e-01]
...
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 7.9795286e-02, -3.7624994e-01, -4.5900899e-01, ...,
2.6527405e-01, 1.6244988e-01, -4.3443482e-02],
[ 7.9840839e-02, -3.7623879e-01, -4.5898870e-01, ...,
6.0503438e-02, -2.9896224e-01, 7.4705563e-02],
[ 7.9717852e-02, -3.7621337e-01, -4.5899495e-01, ...,
-1.8330802e-01, 1.3642092e-01, 2.1550779e-01],
...,
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'resid_adapter': resid_adapterinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[ 7.9795286e-02 -3.7624994e-01 -4.5900899e-01 ... 2.6527405e-01
1.6244988e-01 -4.3443482e-02]
[ 7.9840839e-02 -3.7623879e-01 -4.5898870e-01 ... 6.0503438e-02
-2.9896224e-01 7.4705563e-02]
[ 7.9717852e-02 -3.7621337e-01 -4.5899495e-01 ... -1.8330802e-01
1.3642092e-01 2.1550779e-01]
...
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 7.9795286e-02, -3.7624994e-01, -4.5900899e-01, ...,
2.6527405e-01, 1.6244988e-01, -4.3443482e-02],
[ 7.9840839e-02, -3.7623879e-01, -4.5898870e-01, ...,
6.0503438e-02, -2.9896224e-01, 7.4705563e-02],
[ 7.9717852e-02, -3.7621337e-01, -4.5899495e-01, ...,
-1.8330802e-01, 1.3642092e-01, 2.1550779e-01],
...,
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 16384 parametersw[0,0] 128x0e,128x0e: Array([[ 0.3937284 , 0.06909973, 0.18853222, ..., -0.10455237,
-0.071788 , 0.19546314],
[ 0.61504227, -0.43697816, 0.1261993 , ..., 0.35577676,
0.9733432 , 0.18279058],
[-0.6915464 , 1.3660764 , 0.5243588 , ..., 0.05164947,
0.03794032, 1.1182275 ],
...,
[-0.6308902 , -1.835251 , -0.13734646, ..., 0.76688886,
0.05780329, -0.44389218],
[ 0.06009359, -3.1872697 , 0.38784972, ..., 0.22112429,
-0.03782988, -0.42557165],
[-0.5971681 , -0.8687849 , -0.3203398 , ..., 0.63645226,
-0.05153882, -0.44785336]], dtype=float32):Rendering array...
,}output: 128x0e
[[ 1.8635528e-01 1.8854439e-01 3.1749892e-01 ... -3.5424039e-01
1.6704578e-02 -2.9606482e-01]
[ 1.8637003e-01 1.8862638e-01 3.1749874e-01 ... -3.5427368e-01
1.6694939e-02 -2.9607737e-01]
[ 1.8635909e-01 1.8848734e-01 3.1747714e-01 ... -3.5423234e-01
1.6688688e-02 -2.9606351e-01]
...
[ 2.2420775e-44 7.1466222e-44 0.0000000e+00 ... -3.6433760e-44
-8.4077908e-45 -1.6815582e-44]
[ 2.2420775e-44 7.1466222e-44 0.0000000e+00 ... -3.6433760e-44
-8.4077908e-45 -1.6815582e-44]
[ 2.2420775e-44 7.1466222e-44 0.0000000e+00 ... -3.6433760e-44
-8.4077908e-45 -1.6815582e-44]]key=0,): Array([[ 1.8635528e-01, 1.8854439e-01, 3.1749892e-01, ...,
-3.5424039e-01, 1.6704578e-02, -2.9606482e-01],
[ 1.8637003e-01, 1.8862638e-01, 3.1749874e-01, ...,
-3.5427368e-01, 1.6694939e-02, -2.9607737e-01],
[ 1.8635909e-01, 1.8848734e-01, 3.1747714e-01, ...,
-3.5423234e-01, 1.6688688e-02, -2.9606351e-01],
...,
[ 2.2420775e-44, 7.1466222e-44, 0.0000000e+00, ...,
-3.6433760e-44, -8.4077908e-45, -1.6815582e-44],
[ 2.2420775e-44, 7.1466222e-44, 0.0000000e+00, ...,
-3.6433760e-44, -8.4077908e-45, -1.6815582e-44],
[ 2.2420775e-44, 7.1466222e-44, 0.0000000e+00, ...,
-3.6433760e-44, -8.4077908e-45, -1.6815582e-44]], dtype=float32), :Rendering array...
,,), 'SimpleInteraction_0': SimpleInteraction_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[ 7.9795286e-02 -3.7624994e-01 -4.5900899e-01 ... 2.6527405e-01
1.6244988e-01 -4.3443482e-02]
[ 7.9840839e-02 -3.7623879e-01 -4.5898870e-01 ... 6.0503438e-02
-2.9896224e-01 7.4705563e-02]
[ 7.9717852e-02 -3.7621337e-01 -4.5899495e-01 ... -1.8330802e-01
1.3642092e-01 2.1550779e-01]
...
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 7.9795286e-02, -3.7624994e-01, -4.5900899e-01, ...,
2.6527405e-01, 1.6244988e-01, -4.3443482e-02],
[ 7.9840839e-02, -3.7623879e-01, -4.5898870e-01, ...,
6.0503438e-02, -2.9896224e-01, 7.4705563e-02],
[ 7.9717852e-02, -3.7621337e-01, -4.5899495e-01, ...,
-1.8330802e-01, 1.3642092e-01, 2.1550779e-01],
...,
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'linear_intro': linear_introinput: 'weights_or_input': 128x0e+64x1e+32x2e
[[ 7.9795286e-02 -3.7624994e-01 -4.5900899e-01 ... 2.6527405e-01
1.6244988e-01 -4.3443482e-02]
[ 7.9840839e-02 -3.7623879e-01 -4.5898870e-01 ... 6.0503438e-02
-2.9896224e-01 7.4705563e-02]
[ 7.9717852e-02 -3.7621337e-01 -4.5899495e-01 ... -1.8330802e-01
1.3642092e-01 2.1550779e-01]
...
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-2.8025969e-45 -4.2038954e-45 -2.5223372e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[ 7.9795286e-02, -3.7624994e-01, -4.5900899e-01, ...,
2.6527405e-01, 1.6244988e-01, -4.3443482e-02],
[ 7.9840839e-02, -3.7623879e-01, -4.5898870e-01, ...,
6.0503438e-02, -2.9896224e-01, 7.4705563e-02],
[ 7.9717852e-02, -3.7621337e-01, -4.5899495e-01, ...,
-1.8330802e-01, 1.3642092e-01, 2.1550779e-01],
...,
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-2.8025969e-45, -4.2038954e-45, -2.5223372e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,}params: 21504 parametersw[0,0] 128x0e,128x0e: Array([[ 2.7310824 , 3.0626702 , -2.873125 , ..., -0.46205166,
-2.5823078 , -0.79186493],
[ 1.5819222 , -0.04187162, 0.18020032, ..., -1.421179 ,
0.36783072, -0.41777498],
[ 2.1611493 , -3.5538318 , -3.4319103 , ..., -4.8217306 ,
9.292773 , 1.6516443 ],
...,
[-0.2557063 , -0.08857636, -1.337759 , ..., -0.20876619,
-1.4309722 , -1.1555536 ],
[-0.39878693, 2.476265 , 0.19616641, ..., -0.9218898 ,
1.2366259 , -0.43310922],
[ 0.22334357, 2.3639684 , 1.1262287 , ..., -0.26591426,
0.29230833, -1.8359554 ]], dtype=float32):Rendering array...
, w[1,1] 64x1e,64x1e: Array([[ 2.957393 , -1.2974124 , -1.8927755 , ..., -1.4519069 ,
0.89650595, 2.8739493 ],
[ 2.376374 , -0.32495666, -0.89615333, ..., -0.30789825,
-0.12491573, 1.2192786 ],
[ 1.0807056 , -0.37785774, 1.0011708 , ..., 0.84363145,
0.42042476, -0.62347543],
...,
[-3.6531384 , 1.5399444 , -2.8130946 , ..., 3.2157907 ,
0.9592581 , -1.5591872 ],
[ 3.7486699 , 2.8656209 , -0.21817568, ..., 0.48192185,
0.4373116 , 2.5665076 ],
[ 2.39491 , 2.5794294 , 1.1376072 , ..., -2.5818353 ,
-5.080429 , -0.04517885]], dtype=float32):Rendering array...
, w[2,2] 32x2e,32x2e: Array([[ 1.834679 , -3.5193052 , 0.28963464, ..., 0.7220885 ,
-1.037215 , 1.7926254 ],
[-2.8789775 , 3.5531743 , -1.5598752 , ..., 0.52459097,
1.1114972 , 5.42312 ],
[ 0.54869986, 0.5619909 , -4.4183044 , ..., 0.17171204,
3.3009405 , 0.10807901],
...,
[-2.351306 , -0.04581549, -0.554534 , ..., 2.7257571 ,
0.9586024 , 4.887769 ],
[ 0.21695383, 1.2257326 , -1.1554599 , ..., -0.86416173,
-1.4262117 , 1.0316494 ],
[-3.827227 , 1.203004 , 0.61138934, ..., 0.2602603 ,
-4.1413083 , -3.2767403 ]], dtype=float32):Rendering array...
,}output: 128x0e+64x1e+32x2e
[[-8.4011042e-01 1.1658471e-02 -1.0469003e-01 ... 2.2996256e+00
1.2905813e+00 -9.0730488e-01]
[-8.3996844e-01 1.1436231e-02 -1.0473997e-01 ... 9.0966202e-02
-2.7640491e+00 3.6912921e-01]
[-8.3998084e-01 1.1473636e-02 -1.0464139e-01 ... -1.8454579e+00
1.4720591e+00 1.4872581e+00]
...
[-1.5414283e-44 -6.0255834e-44 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-1.5414283e-44 -6.0255834e-44 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-1.5414283e-44 -6.0255834e-44 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-8.4011042e-01, 1.1658471e-02, -1.0469003e-01, ...,
2.2996256e+00, 1.2905813e+00, -9.0730488e-01],
[-8.3996844e-01, 1.1436231e-02, -1.0473997e-01, ...,
9.0966202e-02, -2.7640491e+00, 3.6912921e-01],
[-8.3998084e-01, 1.1473636e-02, -1.0464139e-01, ...,
-1.8454579e+00, 1.4720591e+00, 1.4872581e+00],
...,
[-1.5414283e-44, -6.0255834e-44, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-1.5414283e-44, -6.0255834e-44, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-1.5414283e-44, -6.0255834e-44, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,,), 'SevenNetConv_0': SevenNetConv_0input: 'vectors': 1x0e+1x1e+1x2e
[[[ 1. 0.38863322 -1.6852407 ... 2.0572233 0.20556238
-0.09173071]
[ 1. 0.7033811 1.0741072 ... 0.1718487 -1.6120846
0.55306447]
[ 1. 0.21102472 -1.4464794 ... 1.2212316 -1.7349366
0.528426 ]
...
[ 1. 0.26149333 1.3177891 ... 0.8235084 1.8597894
0.72726476]
[ 1. 0.8097534 -1.4202117 ... 1.1370419 -1.0489364
-0.211983 ]
[ 1. -0.5494567 1.6424226 ... 1.8979207 0.04951644
-0.19452532]]
[[ 1. 0.3886331 -0.9244453 ... -0.16256315 -1.6854191
1.1898639 ]
[ 1. 0.7033811 -0.46975395 ... -0.8713188 0.91663945
1.1553375 ]
[ 1. 0.21102473 0.0813558 ... -1.1106336 0.18035954
1.8747289 ]
...
[ 1. 0.26149333 1.6056201 ... 1.7642759 -1.232611
0.18411237]
[ 1. 0.8097533 -0.21465297 ... -1.0665194 -0.42010453
1.060244 ]
[ 1. -0.5494566 0.8414356 ... -0.3264505 -1.532431
1.0897164 ]]
[[ 1. 0.38863316 -0.76079535 ... -0.4709053 1.4798577
1.3678855 ]
[ 1. 0.70338124 1.5438613 ... 1.5468084 0.6954451
-0.24076885]
[ 1. 0.21102484 -1.527835 ... 1.4917706 1.554577
0.3722299 ]
...
[ 1. 0.26149336 -0.28783107 ... -1.0254085 -0.62717843
1.7947376 ]
[ 1. 0.8097534 -1.2055587 ... 0.506885 1.4690411
0.15183836]
[ 1. -0.54945654 0.8009872 ... -0.40072533 1.4829144
1.132599 ]]
...
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]
[[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
...
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]
[ 1. 0. 0. ... 0. 0.
0. ]]]key=0,): Array([[[ 1. , 0.38863322, -1.6852407 , ..., 2.0572233 ,
0.20556238, -0.09173071],
[ 1. , 0.7033811 , 1.0741072 , ..., 0.1718487 ,
-1.6120846 , 0.55306447],
[ 1. , 0.21102472, -1.4464794 , ..., 1.2212316 ,
-1.7349366 , 0.528426 ],
...,
[ 1. , 0.26149333, 1.3177891 , ..., 0.8235084 ,
1.8597894 , 0.72726476],
[ 1. , 0.8097534 , -1.4202117 , ..., 1.1370419 ,
-1.0489364 , -0.211983 ],
[ 1. , -0.5494567 , 1.6424226 , ..., 1.8979207 ,
0.04951644, -0.19452532]],
[[ 1. , 0.3886331 , -0.9244453 , ..., -0.16256315,
-1.6854191 , 1.1898639 ],
[ 1. , 0.7033811 , -0.46975395, ..., -0.8713188 ,
0.91663945, 1.1553375 ],
[ 1. , 0.21102473, 0.0813558 , ..., -1.1106336 ,
0.18035954, 1.8747289 ],
...,
[ 1. , 0.26149333, 1.6056201 , ..., 1.7642759 ,
-1.232611 , 0.18411237],
[ 1. , 0.8097533 , -0.21465297, ..., -1.0665194 ,
-0.42010453, 1.060244 ],
[ 1. , -0.5494566 , 0.8414356 , ..., -0.3264505 ,
-1.532431 , 1.0897164 ]],
[[ 1. , 0.38863316, -0.76079535, ..., -0.4709053 ,
1.4798577 , 1.3678855 ],
[ 1. , 0.70338124, 1.5438613 , ..., 1.5468084 ,
0.6954451 , -0.24076885],
[ 1. , 0.21102484, -1.527835 , ..., 1.4917706 ,
1.554577 , 0.3722299 ],
...,
[ 1. , 0.26149336, -0.28783107, ..., -1.0254085 ,
-0.62717843, 1.7947376 ],
[ 1. , 0.8097534 , -1.2055587 , ..., 0.506885 ,
1.4690411 , 0.15183836],
[ 1. , -0.54945654, 0.8009872 , ..., -0.40072533,
1.4829144 , 1.132599 ]],
...,
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]],
[[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
...,
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 1. , 0. , 0. , ..., 0. ,
0. , 0. ]]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'node_feats': 128x0e+64x1e+32x2e
[[-8.4011042e-01 1.1658471e-02 -1.0469003e-01 ... 2.2996256e+00
1.2905813e+00 -9.0730488e-01]
[-8.3996844e-01 1.1436231e-02 -1.0473997e-01 ... 9.0966202e-02
-2.7640491e+00 3.6912921e-01]
[-8.3998084e-01 1.1473636e-02 -1.0464139e-01 ... -1.8454579e+00
1.4720591e+00 1.4872581e+00]
...
[-1.5414283e-44 -6.0255834e-44 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-1.5414283e-44 -6.0255834e-44 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]
[-1.5414283e-44 -6.0255834e-44 1.5414283e-44 ... 0.0000000e+00
0.0000000e+00 0.0000000e+00]]key=0,): Array([[-8.4011042e-01, 1.1658471e-02, -1.0469003e-01, ...,
2.2996256e+00, 1.2905813e+00, -9.0730488e-01],
[-8.3996844e-01, 1.1436231e-02, -1.0473997e-01, ...,
9.0966202e-02, -2.7640491e+00, 3.6912921e-01],
[-8.3998084e-01, 1.1473636e-02, -1.0464139e-01, ...,
-1.8454579e+00, 1.4720591e+00, 1.4872581e+00],
...,
[-1.5414283e-44, -6.0255834e-44, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-1.5414283e-44, -6.0255834e-44, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[-1.5414283e-44, -6.0255834e-44, 1.5414283e-44, ...,
0.0000000e+00, 0.0000000e+00, 0.0000000e+00]], dtype=float32), :Rendering array...
, :Rendering array...
, :Rendering array...
,, 'radial_embedding': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,, 'receivers': Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
, 'avg_num_neighbors': <jax.Array([35.006073], dtype=float32)>:Rendering array...
,}children: 'LazyInMLP_0': LazyInMLP_0input: 'x': 8x0e
[[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932935e-01 -3.58263254e-02 ... 1.72804072e-01
1.32345289e-01 -1.32088214e-01]
[ 1.29668877e-01 7.57186115e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200200e-01 6.40926361e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25451013e-01 ... -1.22329280e-01
1.49454013e-01 2.52312541e-01]
[ 1.38097748e-01 1.78932995e-01 -3.58262807e-02 ... 1.72803968e-01
1.32345453e-01 -1.32087946e-01]
[ 1.29668862e-01 7.57184103e-02 -1.42011866e-01 ... 1.11565115e-02
-1.56199962e-01 6.40934408e-02]
...
[ 1.60929393e-02 -1.55352801e-02 1.57878902e-02 ... -1.23904478e-02
7.83531833e-03 -2.17196788e-03]
[ 1.48273474e-02 -1.42855188e-02 1.45017765e-02 ... -1.11806430e-02
6.84427377e-03 -1.50958879e-03]
[ 1.22505389e-02 -1.17517719e-02 1.18984990e-02 ... -8.80434085e-03
4.97691985e-03 -3.72939656e-04]]
[[ 1.43128961e-01 2.55727351e-01 1.25450939e-01 ... -1.22329280e-01
1.49454191e-01 2.52312481e-01]
[ 1.38097733e-01 1.78932995e-01 -3.58261876e-02 ... 1.72803968e-01
1.32345602e-01 -1.32087946e-01]
[ 1.29668891e-01 7.57185742e-02 -1.42011791e-01 ... 1.11571187e-02
-1.56200111e-01 6.40929043e-02]
...
[ 1.60930213e-02 -1.55353593e-02 1.57879665e-02 ... -1.23905549e-02
7.83538632e-03 -2.17201049e-03]
[ 1.48274256e-02 -1.42855924e-02 1.45018585e-02 ... -1.11807007e-02
6.84436364e-03 -1.50962558e-03]
[ 1.22504616e-02 -1.17516974e-02 1.18984189e-02 ... -8.80426541e-03
4.97684255e-03 -3.72912968e-04]]
...
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]
[[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
...
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]
[ 1.51863381e-01 4.18968886e-01 6.73953354e-01 ... 1.48256779e+00
1.76966441e+00 2.05762792e+00]]]key=0,): Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32), :Rendering array...
,,}children: 'Dense_0': Dense_0input: 'inputs': Array([[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932935e-01, -3.58263254e-02, ...,
1.72804072e-01, 1.32345289e-01, -1.32088214e-01],
[ 1.29668877e-01, 7.57186115e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200200e-01, 6.40926361e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25451013e-01, ...,
-1.22329280e-01, 1.49454013e-01, 2.52312541e-01],
[ 1.38097748e-01, 1.78932995e-01, -3.58262807e-02, ...,
1.72803968e-01, 1.32345453e-01, -1.32087946e-01],
[ 1.29668862e-01, 7.57184103e-02, -1.42011866e-01, ...,
1.11565115e-02, -1.56199962e-01, 6.40934408e-02],
...,
[ 1.60929393e-02, -1.55352801e-02, 1.57878902e-02, ...,
-1.23904478e-02, 7.83531833e-03, -2.17196788e-03],
[ 1.48273474e-02, -1.42855188e-02, 1.45017765e-02, ...,
-1.11806430e-02, 6.84427377e-03, -1.50958879e-03],
[ 1.22505389e-02, -1.17517719e-02, 1.18984990e-02, ...,
-8.80434085e-03, 4.97691985e-03, -3.72939656e-04]],
[[ 1.43128961e-01, 2.55727351e-01, 1.25450939e-01, ...,
-1.22329280e-01, 1.49454191e-01, 2.52312481e-01],
[ 1.38097733e-01, 1.78932995e-01, -3.58261876e-02, ...,
1.72803968e-01, 1.32345602e-01, -1.32087946e-01],
[ 1.29668891e-01, 7.57185742e-02, -1.42011791e-01, ...,
1.11571187e-02, -1.56200111e-01, 6.40929043e-02],
...,
[ 1.60930213e-02, -1.55353593e-02, 1.57879665e-02, ...,
-1.23905549e-02, 7.83538632e-03, -2.17201049e-03],
[ 1.48274256e-02, -1.42855924e-02, 1.45018585e-02, ...,
-1.11807007e-02, 6.84436364e-03, -1.50962558e-03],
[ 1.22504616e-02, -1.17516974e-02, 1.18984189e-02, ...,
-8.80426541e-03, 4.97684255e-03, -3.72912968e-04]],
...,
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]],
[[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
...,
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00],
[ 1.51863381e-01, 4.18968886e-01, 6.73953354e-01, ...,
1.48256779e+00, 1.76966441e+00, 2.05762792e+00]]], dtype=float32):Rendering array...
,}params: 1792 parameterskernel: Array([[ 0.26945972, 0.43817306, 0.03105241, ..., 0.19044442,
0.2351014 , 0.03229111],
[ 0.7108974 , 0.93321556, 0.3339254 , ..., -0.10694128,
0.47263741, 0.99574834],
[ 0.35409585, 1.0157957 , 0.28180015, ..., 0.22329491,
-0.40699825, 1.0774257 ],
...,
[ 0.32090774, -0.531131 , 0.1540587 , ..., -0.4244127 ,
0.31799552, 0.2742479 ],
[ 0.29262787, -0.6745612 , -0.04605831, ..., -0.00225773,
0.02264951, 0.49221578],
[ 0.0800835 , -0.36611298, 0.12036715, ..., 0.13340048,
-0.16945608, 0.12831876]], dtype=float32):Rendering array...
,}output: Array([[[ 1.04371041e-01, 5.70608139e-01, 4.65189153e-03, ...,
9.47358012e-02, -2.08477199e-01, 5.68402112e-01],
[ 9.81239378e-02, 1.99974403e-01, -4.91603874e-02, ...,
-1.46051705e-01, 6.90582171e-02, 2.65027583e-01],
[ 4.36086990e-02, -4.75874059e-02, 2.28474177e-02, ...,
-3.79332565e-02, 1.89851657e-01, -2.00742885e-01],
...,
[-4.51120175e-03, 5.21376915e-03, -5.77478809e-03, ...,
7.74756679e-03, -1.13302087e-02, -3.48034035e-03],
[-4.16124752e-03, 4.81318217e-03, -5.20547898e-03, ...,
7.13169714e-03, -1.04737105e-02, -3.22474353e-03],
[-3.44699319e-03, 3.99698783e-03, -4.09350777e-03, ...,
5.87632274e-03, -8.71221721e-03, -2.69637932e-03]],
[[ 1.04371041e-01, 5.70608139e-01, 4.65189153e-03, ...,
9.47358012e-02, -2.08477199e-01, 5.68402112e-01],
[ 9.81239304e-02, 1.99974477e-01, -4.91604134e-02, ...,
-1.46051615e-01, 6.90580159e-02, 2.65027851e-01],
[ 4.36086841e-02, -4.75880094e-02, 2.28475034e-02, ...,
-3.79327834e-02, 1.89851508e-01, -2.00743139e-01],
...,
[-4.51120175e-03, 5.21376915e-03, -5.77478809e-03, ...,
7.74756679e-03, -1.13302087e-02, -3.48034035e-03],
[-4.16124566e-03, 4.81318729e-03, -5.20544825e-03, ...,
7.13166595e-03, -1.04736676e-02, -3.22475052e-03],
[-3.44700669e-03, 3.99699714e-03, -4.09354130e-03, ...,
5.87636186e-03, -8.71226937e-03, -2.69638211e-03]],
[[ 1.04371056e-01, 5.70607960e-01, 4.65185242e-03, ...,
9.47357640e-02, -2.08477169e-01, 5.68402112e-01],
[ 9.81239602e-02, 1.99974552e-01, -4.91604246e-02, ...,
-1.46051586e-01, 6.90579042e-02, 2.65028059e-01],
[ 4.36087660e-02, -4.75876220e-02, 2.28474680e-02, ...,
-3.79331931e-02, 1.89851642e-01, -2.00742856e-01],
...,
[-4.51123063e-03, 5.21379616e-03, -5.77482814e-03, ...,
7.74761708e-03, -1.13302628e-02, -3.48037179e-03],
[-4.16124752e-03, 4.81318217e-03, -5.20547898e-03, ...,
7.13169714e-03, -1.04737105e-02, -3.22474353e-03],
[-3.44699319e-03, 3.99698783e-03, -4.09350777e-03, ...,
5.87632274e-03, -8.71221721e-03, -2.69637932e-03]],
...,
[[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
...,
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00]],
[[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
...,
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00]],
[[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
...,
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00]]], dtype=float32):Rendering array...
,),}output: 224x0e
[[[ 1.04371041e-01 5.70608139e-01 4.65189153e-03 ... 9.47358012e-02
-2.08477199e-01 5.68402112e-01]
[ 9.81239378e-02 1.99974403e-01 -4.91603874e-02 ... -1.46051705e-01
6.90582171e-02 2.65027583e-01]
[ 4.36086990e-02 -4.75874059e-02 2.28474177e-02 ... -3.79332565e-02
1.89851657e-01 -2.00742885e-01]
...
[-4.51120175e-03 5.21376915e-03 -5.77478809e-03 ... 7.74756679e-03
-1.13302087e-02 -3.48034035e-03]
[-4.16124752e-03 4.81318217e-03 -5.20547898e-03 ... 7.13169714e-03
-1.04737105e-02 -3.22474353e-03]
[-3.44699319e-03 3.99698783e-03 -4.09350777e-03 ... 5.87632274e-03
-8.71221721e-03 -2.69637932e-03]]
[[ 1.04371041e-01 5.70608139e-01 4.65189153e-03 ... 9.47358012e-02
-2.08477199e-01 5.68402112e-01]
[ 9.81239304e-02 1.99974477e-01 -4.91604134e-02 ... -1.46051615e-01
6.90580159e-02 2.65027851e-01]
[ 4.36086841e-02 -4.75880094e-02 2.28475034e-02 ... -3.79327834e-02
1.89851508e-01 -2.00743139e-01]
...
[-4.51120175e-03 5.21376915e-03 -5.77478809e-03 ... 7.74756679e-03
-1.13302087e-02 -3.48034035e-03]
[-4.16124566e-03 4.81318729e-03 -5.20544825e-03 ... 7.13166595e-03
-1.04736676e-02 -3.22475052e-03]
[-3.44700669e-03 3.99699714e-03 -4.09354130e-03 ... 5.87636186e-03
-8.71226937e-03 -2.69638211e-03]]
[[ 1.04371056e-01 5.70607960e-01 4.65185242e-03 ... 9.47357640e-02
-2.08477169e-01 5.68402112e-01]
[ 9.81239602e-02 1.99974552e-01 -4.91604246e-02 ... -1.46051586e-01
6.90579042e-02 2.65028059e-01]
[ 4.36087660e-02 -4.75876220e-02 2.28474680e-02 ... -3.79331931e-02
1.89851642e-01 -2.00742856e-01]
...
[-4.51123063e-03 5.21379616e-03 -5.77482814e-03 ... 7.74761708e-03
-1.13302628e-02 -3.48037179e-03]
[-4.16124752e-03 4.81318217e-03 -5.20547898e-03 ... 7.13169714e-03
-1.04737105e-02 -3.22474353e-03]
[-3.44699319e-03 3.99698783e-03 -4.09350777e-03 ... 5.87632274e-03
-8.71221721e-03 -2.69637932e-03]]
...
[[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
...
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]]
[[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
...
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]]
[[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
...
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]
[ 2.77915382e+00 -3.00344849e+00 1.46300197e+00 ... -4.39159647e-02
1.38193977e+00 2.21181726e+00]]]key=0,): Array([[[ 1.04371041e-01, 5.70608139e-01, 4.65189153e-03, ...,
9.47358012e-02, -2.08477199e-01, 5.68402112e-01],
[ 9.81239378e-02, 1.99974403e-01, -4.91603874e-02, ...,
-1.46051705e-01, 6.90582171e-02, 2.65027583e-01],
[ 4.36086990e-02, -4.75874059e-02, 2.28474177e-02, ...,
-3.79332565e-02, 1.89851657e-01, -2.00742885e-01],
...,
[-4.51120175e-03, 5.21376915e-03, -5.77478809e-03, ...,
7.74756679e-03, -1.13302087e-02, -3.48034035e-03],
[-4.16124752e-03, 4.81318217e-03, -5.20547898e-03, ...,
7.13169714e-03, -1.04737105e-02, -3.22474353e-03],
[-3.44699319e-03, 3.99698783e-03, -4.09350777e-03, ...,
5.87632274e-03, -8.71221721e-03, -2.69637932e-03]],
[[ 1.04371041e-01, 5.70608139e-01, 4.65189153e-03, ...,
9.47358012e-02, -2.08477199e-01, 5.68402112e-01],
[ 9.81239304e-02, 1.99974477e-01, -4.91604134e-02, ...,
-1.46051615e-01, 6.90580159e-02, 2.65027851e-01],
[ 4.36086841e-02, -4.75880094e-02, 2.28475034e-02, ...,
-3.79327834e-02, 1.89851508e-01, -2.00743139e-01],
...,
[-4.51120175e-03, 5.21376915e-03, -5.77478809e-03, ...,
7.74756679e-03, -1.13302087e-02, -3.48034035e-03],
[-4.16124566e-03, 4.81318729e-03, -5.20544825e-03, ...,
7.13166595e-03, -1.04736676e-02, -3.22475052e-03],
[-3.44700669e-03, 3.99699714e-03, -4.09354130e-03, ...,
5.87636186e-03, -8.71226937e-03, -2.69638211e-03]],
[[ 1.04371056e-01, 5.70607960e-01, 4.65185242e-03, ...,
9.47357640e-02, -2.08477169e-01, 5.68402112e-01],
[ 9.81239602e-02, 1.99974552e-01, -4.91604246e-02, ...,
-1.46051586e-01, 6.90579042e-02, 2.65028059e-01],
[ 4.36087660e-02, -4.75876220e-02, 2.28474680e-02, ...,
-3.79331931e-02, 1.89851642e-01, -2.00742856e-01],
...,
[-4.51123063e-03, 5.21379616e-03, -5.77482814e-03, ...,
7.74761708e-03, -1.13302628e-02, -3.48037179e-03],
[-4.16124752e-03, 4.81318217e-03, -5.20547898e-03, ...,
7.13169714e-03, -1.04737105e-02, -3.22474353e-03],
[-3.44699319e-03, 3.99698783e-03, -4.09350777e-03, ...,
5.87632274e-03, -8.71221721e-03, -2.69637932e-03]],
...,
[[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
...,
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00]],
[[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
...,
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00]],
[[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
...,
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00],
[ 2.77915382e+00, -3.00344849e+00, 1.46300197e+00, ...,
-4.39159647e-02, 1.38193977e+00, 2.21181726e+00]]], dtype=float32), :Rendering array...
,,),}output: 224x0e
[[ 0.00023087 0.08408103 0.00557164 ... 0.0128152 0.03508127
-0.08928271]
[ 0.00022966 0.08408215 0.00557262 ... 0.01280922 0.03514008
-0.0893766 ]
[ 0.00022617 0.08408467 0.00557321 ... 0.01279275 0.03514772
-0.08937456]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.00023087, 0.08408103, 0.00557164, ..., 0.0128152 ,
0.03508127, -0.08928271],
[ 0.00022966, 0.08408215, 0.00557262, ..., 0.01280922,
0.03514008, -0.0893766 ],
[ 0.00022617, 0.08408467, 0.00557321, ..., 0.01279275,
0.03514772, -0.08937456],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
,,), 'linear_outro': linear_outroinput: 'weights_or_input': 224x0e
[[ 0.00023087 0.08408103 0.00557164 ... 0.0128152 0.03508127
-0.08928271]
[ 0.00022966 0.08408215 0.00557262 ... 0.01280922 0.03514008
-0.0893766 ]
[ 0.00022617 0.08408467 0.00557321 ... 0.01279275 0.03514772
-0.08937456]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[ 0.00023087, 0.08408103, 0.00557164, ..., 0.0128152 ,
0.03508127, -0.08928271],
[ 0.00022966, 0.08408215, 0.00557262, ..., 0.01280922,
0.03514008, -0.0893766 ],
[ 0.00022617, 0.08408467, 0.00557321, ..., 0.01279275,
0.03514772, -0.08937456],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
,,}params: 28672 parametersw[0,0] 224x0e,128x0e: Array([[-1.0418724 , -0.41554564, -0.68577147, ..., -0.3166389 ,
0.8383788 , 0.1606412 ],
[-1.3690069 , 1.2751813 , 0.5149619 , ..., 3.0915513 ,
0.19957714, -2.5659294 ],
[ 0.9293653 , 1.6508423 , 0.77863246, ..., 0.11865105,
0.87970626, -0.9510862 ],
...,
[ 1.1623504 , -0.6409943 , -0.52440107, ..., 0.7576407 ,
-1.0330133 , -1.2527919 ],
[ 0.48919797, 1.302377 , -0.38269427, ..., 0.03654883,
0.9190065 , 1.2383986 ],
[-0.7033828 , -0.35301045, -0.9827658 , ..., 0.5619353 ,
-0.15537892, 0.10997062]], dtype=float32):Rendering array...
,}output: 128x0e
[[-0.08303457 0.18125378 -0.11307325 ... -0.19846307 0.07019387
-0.05175884]
[-0.08301673 0.1812089 -0.11308573 ... -0.19842891 0.07021064
-0.0517609 ]
[-0.08301759 0.18124628 -0.11304977 ... -0.19843075 0.07020815
-0.05179232]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.08303457, 0.18125378, -0.11307325, ..., -0.19846307,
0.07019387, -0.05175884],
[-0.08301673, 0.1812089 , -0.11308573, ..., -0.19842891,
0.07021064, -0.0517609 ],
[-0.08301759, 0.18124628, -0.11304977, ..., -0.19843075,
0.07020815, -0.05179232],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
,,),}output: 128x0e
[[-0.08303457 0.18125378 -0.11307325 ... -0.19846307 0.07019387
-0.05175884]
[-0.08301673 0.1812089 -0.11308573 ... -0.19842891 0.07021064
-0.0517609 ]
[-0.08301759 0.18124628 -0.11304977 ... -0.19843075 0.07020815
-0.05179232]
...
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]
[ 0. 0. 0. ... 0. 0.
0. ]]key=0,): Array([[-0.08303457, 0.18125378, -0.11307325, ..., -0.19846307,
0.07019387, -0.05175884],
[-0.08301673, 0.1812089 , -0.11308573, ..., -0.19842891,
0.07021064, -0.0517609 ],
[-0.08301759, 0.18124628, -0.11304977, ..., -0.19843075,
0.07020815, -0.05179232],
...,
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ],
[ 0. , 0. , 0. , ..., 0. ,
0. , 0. ]], dtype=float32), :Rendering array...
,,),}output: 128x0e
[[ 1.03320710e-01 3.69798183e-01 2.04425663e-01 ... -5.52703440e-01
8.68984461e-02 -3.47823679e-01]
[ 1.03353299e-01 3.69835258e-01 2.04413012e-01 ... -5.52702606e-01
8.69055837e-02 -3.47838283e-01]
[ 1.03341505e-01 3.69733632e-01 2.04427361e-01 ... -5.52663088e-01
8.68968368e-02 -3.47855836e-01]
...
[ 2.24207754e-44 7.14662217e-44 0.00000000e+00 ... -3.64337601e-44
-8.40779079e-45 -1.68155816e-44]
[ 2.24207754e-44 7.14662217e-44 0.00000000e+00 ... -3.64337601e-44
-8.40779079e-45 -1.68155816e-44]
[ 2.24207754e-44 7.14662217e-44 0.00000000e+00 ... -3.64337601e-44
-8.40779079e-45 -1.68155816e-44]]key=0,): Array([[ 1.03320710e-01, 3.69798183e-01, 2.04425663e-01, ...,
-5.52703440e-01, 8.68984461e-02, -3.47823679e-01],
[ 1.03353299e-01, 3.69835258e-01, 2.04413012e-01, ...,
-5.52702606e-01, 8.69055837e-02, -3.47838283e-01],
[ 1.03341505e-01, 3.69733632e-01, 2.04427361e-01, ...,
-5.52663088e-01, 8.68968368e-02, -3.47855836e-01],
...,
[ 2.24207754e-44, 7.14662217e-44, 0.00000000e+00, ...,
-3.64337601e-44, -8.40779079e-45, -1.68155816e-44],
[ 2.24207754e-44, 7.14662217e-44, 0.00000000e+00, ...,
-3.64337601e-44, -8.40779079e-45, -1.68155816e-44],
[ 2.24207754e-44, 7.14662217e-44, 0.00000000e+00, ...,
-3.64337601e-44, -8.40779079e-45, -1.68155816e-44]], dtype=float32), :Rendering array...
,,), 'self_connection': self_connectioninput: 'node_feats': 128x0e
[[ 1.03320710e-01 3.69798183e-01 2.04425663e-01 ... -5.52703440e-01
8.68984461e-02 -3.47823679e-01]
[ 1.03353299e-01 3.69835258e-01 2.04413012e-01 ... -5.52702606e-01
8.69055837e-02 -3.47838283e-01]
[ 1.03341505e-01 3.69733632e-01 2.04427361e-01 ... -5.52663088e-01
8.68968368e-02 -3.47855836e-01]
...
[ 2.24207754e-44 7.14662217e-44 0.00000000e+00 ... -3.64337601e-44
-8.40779079e-45 -1.68155816e-44]
[ 2.24207754e-44 7.14662217e-44 0.00000000e+00 ... -3.64337601e-44
-8.40779079e-45 -1.68155816e-44]
[ 2.24207754e-44 7.14662217e-44 0.00000000e+00 ... -3.64337601e-44
-8.40779079e-45 -1.68155816e-44]]key=0,): Array([[ 1.03320710e-01, 3.69798183e-01, 2.04425663e-01, ...,
-5.52703440e-01, 8.68984461e-02, -3.47823679e-01],
[ 1.03353299e-01, 3.69835258e-01, 2.04413012e-01, ...,
-5.52702606e-01, 8.69055837e-02, -3.47838283e-01],
[ 1.03341505e-01, 3.69733632e-01, 2.04427361e-01, ...,
-5.52663088e-01, 8.68968368e-02, -3.47855836e-01],
...,
[ 2.24207754e-44, 7.14662217e-44, 0.00000000e+00, ...,
-3.64337601e-44, -8.40779079e-45, -1.68155816e-44],
[ 2.24207754e-44, 7.14662217e-44, 0.00000000e+00, ...,
-3.64337601e-44, -8.40779079e-45, -1.68155816e-44],
[ 2.24207754e-44, 7.14662217e-44, 0.00000000e+00, ...,
-3.64337601e-44, -8.40779079e-45, -1.68155816e-44]], dtype=float32), :Rendering array...
,, 'node_specie': Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, 'species_embed': Array([[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
[ 4.2173825e-03, -7.3911045e-03, -1.3092387e-02, ...,
6.2542468e-01, 1.6457355e-02, -3.5454072e-02],
...,
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45],
[-4.2038954e-45, -4.2038954e-45, -4.2038954e-45, ...,
-4.2038954e-45, 4.2038954e-45, -4.2038954e-45]], dtype=float32):Rendering array...
,}output: 128x0e
[[ 9.1082357e-02 3.6666879e-01 1.8882154e-01 ... -3.3844477e-01
7.6008528e-02 -2.4137130e-01]
[ 9.1112502e-02 3.6671111e-01 1.8880878e-01 ... -3.3844444e-01
7.6015025e-02 -2.4137941e-01]
[ 9.1101587e-02 3.6659512e-01 1.8882325e-01 ... -3.3842871e-01
7.6007061e-02 -2.4138913e-01]
...
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]key=0,): Array([[ 9.1082357e-02, 3.6666879e-01, 1.8882154e-01, ...,
-3.3844477e-01, 7.6008528e-02, -2.4137130e-01],
[ 9.1112502e-02, 3.6671111e-01, 1.8880878e-01, ...,
-3.3844444e-01, 7.6015025e-02, -2.4137941e-01],
[ 9.1101587e-02, 3.6659512e-01, 1.8882325e-01, ...,
-3.3842871e-01, 7.6007061e-02, -2.4138913e-01],
...,
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]], dtype=float32), :Rendering array...
,,), 'readout': readoutinput: 'x': 128x0e
[[ 9.1082357e-02 3.6666879e-01 1.8882154e-01 ... -3.3844477e-01
7.6008528e-02 -2.4137130e-01]
[ 9.1112502e-02 3.6671111e-01 1.8880878e-01 ... -3.3844444e-01
7.6015025e-02 -2.4137941e-01]
[ 9.1101587e-02 3.6659512e-01 1.8882325e-01 ... -3.3842871e-01
7.6007061e-02 -2.4138913e-01]
...
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]key=0,): Array([[ 9.1082357e-02, 3.6666879e-01, 1.8882154e-01, ...,
-3.3844477e-01, 7.6008528e-02, -2.4137130e-01],
[ 9.1112502e-02, 3.6671111e-01, 1.8880878e-01, ...,
-3.3844444e-01, 7.6015025e-02, -2.4137941e-01],
[ 9.1101587e-02, 3.6659512e-01, 1.8882325e-01, ...,
-3.3842871e-01, 7.6007061e-02, -2.4138913e-01],
...,
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]], dtype=float32), :Rendering array...
,,}output: 128x0e
[[ 9.1082357e-02 3.6666879e-01 1.8882154e-01 ... -3.3844477e-01
7.6008528e-02 -2.4137130e-01]
[ 9.1112502e-02 3.6671111e-01 1.8880878e-01 ... -3.3844444e-01
7.6015025e-02 -2.4137941e-01]
[ 9.1101587e-02 3.6659512e-01 1.8882325e-01 ... -3.3842871e-01
7.6007061e-02 -2.4138913e-01]
...
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]key=0,): Array([[ 9.1082357e-02, 3.6666879e-01, 1.8882154e-01, ...,
-3.3844477e-01, 7.6008528e-02, -2.4137130e-01],
[ 9.1112502e-02, 3.6671111e-01, 1.8880878e-01, ...,
-3.3844444e-01, 7.6015025e-02, -2.4137941e-01],
[ 9.1101587e-02, 3.6659512e-01, 1.8882325e-01, ...,
-3.3842871e-01, 7.6007061e-02, -2.4138913e-01],
...,
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]], dtype=float32), :Rendering array...
,,),}output: 128x0e
[[ 9.1082357e-02 3.6666879e-01 1.8882154e-01 ... -3.3844477e-01
7.6008528e-02 -2.4137130e-01]
[ 9.1112502e-02 3.6671111e-01 1.8880878e-01 ... -3.3844444e-01
7.6015025e-02 -2.4137941e-01]
[ 9.1101587e-02 3.6659512e-01 1.8882325e-01 ... -3.3842871e-01
7.6007061e-02 -2.4138913e-01]
...
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]key=0,): Array([[ 9.1082357e-02, 3.6666879e-01, 1.8882154e-01, ...,
-3.3844477e-01, 7.6008528e-02, -2.4137130e-01],
[ 9.1112502e-02, 3.6671111e-01, 1.8880878e-01, ...,
-3.3844444e-01, 7.6015025e-02, -2.4137941e-01],
[ 9.1101587e-02, 3.6659512e-01, 1.8882325e-01, ...,
-3.3842871e-01, 7.6007061e-02, -2.4138913e-01],
...,
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]], dtype=float32), :Rendering array...
,, 128x0e
[[ 9.1082357e-02 3.6666879e-01 1.8882154e-01 ... -3.3844477e-01
7.6008528e-02 -2.4137130e-01]
[ 9.1112502e-02 3.6671111e-01 1.8880878e-01 ... -3.3844444e-01
7.6015025e-02 -2.4137941e-01]
[ 9.1101587e-02 3.6659512e-01 1.8882325e-01 ... -3.3842871e-01
7.6007061e-02 -2.4138913e-01]
...
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]
[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]key=0,): Array([[ 9.1082357e-02, 3.6666879e-01, 1.8882154e-01, ...,
-3.3844477e-01, 7.6008528e-02, -2.4137130e-01],
[ 9.1112502e-02, 3.6671111e-01, 1.8880878e-01, ...,
-3.3844444e-01, 7.6015025e-02, -2.4137941e-01],
[ 9.1101587e-02, 3.6659512e-01, 1.8882325e-01, ...,
-3.3842871e-01, 7.6007061e-02, -2.4138913e-01],
...,
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]], dtype=float32), :Rendering array...
,,),),}output: 128x0e
[[[ 9.1082357e-02 3.6666879e-01 1.8882154e-01 ... -3.3844477e-01
7.6008528e-02 -2.4137130e-01]]
[[ 9.1112502e-02 3.6671111e-01 1.8880878e-01 ... -3.3844444e-01
7.6015025e-02 -2.4137941e-01]]
[[ 9.1101587e-02 3.6659512e-01 1.8882325e-01 ... -3.3842871e-01
7.6007061e-02 -2.4138913e-01]]
...
[[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]
[[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]
[[ 1.8216880e-44 6.1657132e-44 0.0000000e+00 ... -3.0828566e-44
-7.0064923e-45 -1.4012985e-44]]]key=0,): Array([[[ 9.1082357e-02, 3.6666879e-01, 1.8882154e-01, ...,
-3.3844477e-01, 7.6008528e-02, -2.4137130e-01]],
[[ 9.1112502e-02, 3.6671111e-01, 1.8880878e-01, ...,
-3.3844444e-01, 7.6015025e-02, -2.4137941e-01]],
[[ 9.1101587e-02, 3.6659512e-01, 1.8882325e-01, ...,
-3.3842871e-01, 7.6007061e-02, -2.4138913e-01]],
...,
[[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]],
[[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]],
[[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]]], dtype=float32), :Rendering array...
,,), 'head_dropout': head_dropoutinput: 'inputs': Array([[ 9.1082357e-02, 3.6666879e-01, 1.8882154e-01, ...,
-3.3844477e-01, 7.6008528e-02, -2.4137130e-01],
[ 9.1112502e-02, 3.6671111e-01, 1.8880878e-01, ...,
-3.3844444e-01, 7.6015025e-02, -2.4137941e-01],
[ 9.1101587e-02, 3.6659512e-01, 1.8882325e-01, ...,
-3.3842871e-01, 7.6007061e-02, -2.4138913e-01],
...,
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44],
[ 1.8216880e-44, 6.1657132e-44, 0.0000000e+00, ...,
-3.0828566e-44, -7.0064923e-45, -1.4012985e-44]], dtype=float32):Rendering array...
, 'deterministic': False,}output: Array([[ 1.0120262e-01, 4.0740979e-01, 0.0000000e+00, ...,
-3.7604976e-01, 8.4453925e-02, -2.6819035e-01],
[ 1.0123612e-01, 4.0745682e-01, 2.0978755e-01, ...,
-3.7604940e-01, 8.4461145e-02, -2.6819935e-01],
[ 1.0122399e-01, 4.0732792e-01, 2.0980363e-01, ...,
-3.7603191e-01, 8.4452294e-02, -2.6821014e-01],
...,
[ 1.9618179e-44, 0.0000000e+00, 0.0000000e+00, ...,
-3.3631163e-44, 0.0000000e+00, -1.5414283e-44],
[ 1.9618179e-44, 6.8663625e-44, 0.0000000e+00, ...,
-3.3631163e-44, -8.4077908e-45, 0.0000000e+00],
[ 1.9618179e-44, 6.8663625e-44, 0.0000000e+00, ...,
-3.3631163e-44, -8.4077908e-45, -1.5414283e-44]], dtype=float32):Rendering array...
,), 'head': headinput: 'x': Array([[ 1.0120262e-01, 4.0740979e-01, 0.0000000e+00, ...,
-3.7604976e-01, 8.4453925e-02, -2.6819035e-01],
[ 1.0123612e-01, 4.0745682e-01, 2.0978755e-01, ...,
-3.7604940e-01, 8.4461145e-02, -2.6819935e-01],
[ 1.0122399e-01, 4.0732792e-01, 2.0980363e-01, ...,
-3.7603191e-01, 8.4452294e-02, -2.6821014e-01],
...,
[ 1.9618179e-44, 0.0000000e+00, 0.0000000e+00, ...,
-3.3631163e-44, 0.0000000e+00, -1.5414283e-44],
[ 1.9618179e-44, 6.8663625e-44, 0.0000000e+00, ...,
-3.3631163e-44, -8.4077908e-45, 0.0000000e+00],
[ 1.9618179e-44, 6.8663625e-44, 0.0000000e+00, ...,
-3.3631163e-44, -8.4077908e-45, -1.5414283e-44]], dtype=float32):Rendering array...
,}children: 'Dense_0': Dense_0input: 'inputs': Array([[ 1.0120262e-01, 4.0740979e-01, 0.0000000e+00, ...,
-3.7604976e-01, 8.4453925e-02, -2.6819035e-01],
[ 1.0123612e-01, 4.0745682e-01, 2.0978755e-01, ...,
-3.7604940e-01, 8.4461145e-02, -2.6819935e-01],
[ 1.0122399e-01, 4.0732792e-01, 2.0980363e-01, ...,
-3.7603191e-01, 8.4452294e-02, -2.6821014e-01],
...,
[ 1.9618179e-44, 0.0000000e+00, 0.0000000e+00, ...,
-3.3631163e-44, 0.0000000e+00, -1.5414283e-44],
[ 1.9618179e-44, 6.8663625e-44, 0.0000000e+00, ...,
-3.3631163e-44, -8.4077908e-45, 0.0000000e+00],
[ 1.9618179e-44, 6.8663625e-44, 0.0000000e+00, ...,
-3.3631163e-44, -8.4077908e-45, -1.5414283e-44]], dtype=float32):Rendering array...
,}params: 128 parameterskernel: Array([[-7.3323093e-02],
[ 2.0857520e-02],
[-7.4718341e-02],
[-6.2562965e-02],
[ 4.9506143e-02],
[ 6.4088538e-02],
[-2.1354200e-02],
[ 4.1273855e-02],
[-8.7901473e-02],
[-7.6430246e-02],
[-5.6931920e-02],
[-7.9819717e-02],
[ 4.6707336e-02],
[-1.0819524e-03],
[-4.9101446e-02],
[-4.7692545e-03],
[-6.8938918e-02],
[ 4.4658501e-02],
[ 5.2118856e-02],
[ 5.2286707e-02],
[-2.1002230e-03],
[-1.1098199e-03],
[ 6.4877555e-02],
[-5.3715833e-02],
[ 5.3442709e-02],
[-1.5025394e-03],
[ 5.8496479e-02],
[-6.8124473e-02],
[-7.5792834e-02],
[-7.0392489e-02],
[ 2.5195791e-03],
[-6.4322315e-02],
[-8.0225900e-02],
[ 5.1826373e-02],
[-6.8810103e-03],
[ 5.3192981e-02],
[-7.2647415e-02],
[-5.9058398e-02],
[-6.0489949e-02],
[ 5.6477930e-02],
[ 2.9234491e-02],
[-7.9039140e-03],
[ 1.3872646e-04],
[-6.3473918e-02],
[ 5.2041385e-02],
[ 4.2663690e-02],
[-6.8757080e-02],
[-7.6088086e-02],
[-4.5156661e-02],
[-3.1352520e-02],
[-7.5085357e-02],
[ 4.7567796e-02],
[-6.4098891e-03],
[ 2.0447827e-03],
[-6.5735050e-02],
[ 5.3788118e-02],
[-3.9345171e-02],
[-5.1456850e-02],
[ 5.3774357e-02],
[ 2.8405490e-03],
[-5.4608274e-05],
[-8.2065895e-02],
[-8.2310170e-02],
[-5.8823891e-02],
[ 4.0307153e-02],
[ 4.6815358e-02],
[-6.5950818e-02],
[ 5.0985575e-02],
[ 5.9336908e-02],
[-2.6204260e-03],
[-5.7491747e-04],
[-6.8728998e-02],
[-7.5307488e-02],
[ 4.6691045e-02],
[-8.5048757e-02],
[-7.2264649e-02],
[-2.3038473e-02],
[-7.1395472e-02],
[ 3.2673776e-03],
[ 4.2771209e-02],
[ 5.5624627e-02],
[-2.1413467e-03],
[-4.0388420e-02],
[ 1.2444846e-03],
[-5.2980776e-04],
[ 2.5609902e-03],
[ 3.7861802e-02],
[ 5.5019349e-02],
[-2.2600594e-03],
[-5.9719913e-02],
[-5.6984358e-02],
[ 4.7067355e-02],
[-2.2469580e-03],
[-6.7480549e-02],
[-9.1918424e-02],
[-5.4613639e-02],
[ 4.6480380e-02],
[-6.2892988e-02],
[ 5.2705746e-02],
[-7.3074378e-02],
[ 4.4752839e-03],
[ 5.2366756e-02],
[-3.6038421e-02],
[ 6.0057804e-02],
[ 4.4242196e-02],
[ 4.7954105e-02],
[-1.8862605e-03],
[ 5.4067023e-02],
[-7.5417444e-02],
[-3.2411609e-03],
[-7.7518038e-02],
[-8.6692937e-02],
[-5.1371388e-02],
[-3.8555808e-02],
[ 3.9256103e-02],
[ 6.9600618e-03],
[ 4.0010110e-02],
[-3.0875580e-02],
[-6.9769055e-02],
[-1.2040902e-03],
[-6.6607162e-02],
[-2.6735226e-03],
[ 4.8590209e-02],
[ 4.9908720e-02],
[ 4.9790747e-02],
[ 5.8867864e-02],
[-7.4486784e-04],
[ 3.8053673e-02]], dtype=float32):Rendering array...
,}output: Array([[-1.3220434e+00],
[-1.3318594e+00],
[-1.3613300e+00],
...,
[-5.4650640e-44],
[-5.6051939e-44],
[-5.0446745e-44]], dtype=float32):Rendering array...
,),}output: Array([[-1.3220434e+00],
[-1.3318594e+00],
[-1.3613300e+00],
...,
[-5.4650640e-44],
[-5.6051939e-44],
[-5.0446745e-44]], dtype=float32):Rendering array...
,), 'rescale': rescaleinput: 'cg': nodes=species=Array([4, 4, 4, ..., 0, 0, 0], dtype=int16):Rendering array...
, cart=Array([[-1.6074809 , 3.990498 , 0.59335387],
[13.654599 , 0.6031299 , 4.7416563 ],
[ 2.6521325 , 3.3873684 , 4.7416563 ],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]], dtype=float32):Rendering array...
, graph_i=Array([ 0, 0, 0, ..., 31, 31, 31], dtype=int32):Rendering array...
,), edges=to_jimage=Array([[[-1, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, -1]],
[[ 0, -1, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 1, 0, 0],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]],
...,
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]],
[[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0],
...,
[ 0, 0, 0],
[ 0, 0, 0],
[ 0, 0, 0]]], dtype=int8):Rendering array...
, receiver=Array([[113, 150, 49, ..., 162, 136, 72],
[115, 151, 51, ..., 163, 137, 73],
[112, 152, 48, ..., 164, 137, 74],
...,
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936],
[936, 936, 936, ..., 936, 936, 936]], dtype=uint32):Rendering array...
,), n_node=Array([168, 80, 68, 60, 53, 48, 42, 40, 36, 32, 31, 28, 26,
24, 24, 21, 20, 18, 16, 16, 14, 12, 12, 10, 9, 8,
6, 5, 4, 4, 1, 88], dtype=uint16):Rendering array...
, padding_mask=Array([ True, True, True, True, True, True, True, True, True,
True, True, True, True, True, True, True, True, True,
True, True, True, True, True, True, True, True, True,
True, True, True, True, False], dtype=bool):Rendering array...
, graph_data=dataset_id=Array([ 114536034, 1780206004, -1687275888, 1202003, 1758626004,
1556291002, 139460000, 1554815003, 118640006, 1675109005,
-1651451869, 1555003018, 1615850000, -1709938884, -1592934888,
-1650297867, -1696810885, -1663817880, -1696683861, -1710558886,
-1662911874, -1781222858, 123675020, -1806109888, -1804816886,
-1875814888, -1668181878, 122133000, 1864969002, -1700082886,
120483000, 0], dtype=int32):Rendering array...
, abc=Array([[17.914213 , 17.914213 , 8.296605 ],
[10.275203 , 16.84246 , 5.0821056],
[ 6.74365 , 9.197391 , 12.196357 ],
[ 9.433409 , 9.433409 , 20.947056 ],
[10.816606 , 10.426013 , 10.417979 ],
[ 6.8768296, 11.247085 , 19.015215 ],
[ 9.827564 , 9.827564 , 6.762959 ],
[ 6.386287 , 6.386287 , 12.2509365],
[ 8.614349 , 8.614349 , 8.614349 ],
[ 7.9092207, 7.7653685, 6.9275327],
[ 8.073368 , 20.393888 , 10.60539 ],
[ 5.306318 , 9.112615 , 10.040712 ],
[ 5.627906 , 6.953792 , 10.501416 ],
[ 5.037883 , 5.8846297, 7.8679 ],
[ 5.66 , 5.650074 , 9.156885 ],
[ 6.1716037, 8.621225 , 5.767166 ],
[ 6.2029986, 8.613454 , 8.613454 ],
[ 5.494526 , 5.494526 , 7.4353304],
[ 3.7546806, 4.811737 , 5.4126735],
[ 2.715433 , 5.2583437, 9.388806 ],
[ 6.0365143, 6.034208 , 6.02813 ],
[ 5.419802 , 6.49084 , 6.660221 ],
[ 4.117953 , 5.849566 , 5.849566 ],
[ 4.299559 , 4.299559 , 10.174184 ],
[ 4.235417 , 7.5434947, 7.5435066],
[ 6.565039 , 6.565039 , 6.565039 ],
[ 4.3139734, 5.5680494, 5.484443 ],
[ 5.9383936, 5.9383936, 5.9383936],
[ 4.4320164, 4.4320164, 4.4320164],
[ 3.640625 , 3.640625 , 6.0473022],
[ 3.571267 , 3.571267 , 3.571267 ],
[ 0. , 0. , 0. ]], dtype=float32):Rendering array...
, angles_rad=Array([[1.5707964, 1.5707964, 2.0943952],
[1.5707964, 1.5707964, 1.5707964],
[1.2343053, 1.5818529, 1.1629694],
[1.5707964, 1.5707964, 1.5707964],
[1.8930188, 1.8607063, 2.0151982],
[1.7593405, 1.5707964, 1.5707964],
[1.5707964, 1.5707964, 2.0943952],
[1.730846 , 1.730846 , 1.8141689],
[1.0478324, 1.0478324, 1.0478324],
[1.8424883, 1.5707964, 1.5707964],
[0.3288564, 1.0035257, 1.0851887],
[1.5707964, 1.5707964, 1.5707964],
[1.5707964, 1.5707964, 1.5707964],
[1.7750735, 1.7689804, 1.2668823],
[1.6448978, 1.4945464, 1.8871104],
[1.5798758, 1.5553586, 1.5304748],
[1.2806493, 1.2024467, 1.2024467],
[1.2456238, 1.397332 , 1.9114691],
[1.5707964, 1.5707964, 1.5707964],
[1.644386 , 1.5707964, 1.5707964],
[1.5746922, 2.094859 , 1.0429444],
[2.0565953, 1.3555512, 1.5138267],
[1.5707964, 1.5707964, 1.5707964],
[1.5707964, 1.5707964, 1.5707964],
[2.0943942, 1.5707964, 1.5707964],
[2.2965217, 2.2965217, 1.237292 ],
[1.6661328, 1.9750055, 1.1730014],
[2.4071865, 2.4071865, 1.0650531],
[1.0471976, 1.0471976, 1.0471976],
[1.5707964, 1.5707964, 2.0943952],
[1.0471976, 1.0471976, 1.0471976],
[0. , 0. , 0. ]], dtype=float32):Rendering array...
, lat=Array([[[ 1.7914213e+01, 1.5000001e-07, 0.0000000e+00],
[-8.9571066e+00, 1.5514163e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 8.2966051e+00]],
[[ 1.0275203e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.6842461e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 5.0821056e+00]],
[[ 1.6343699e-01, -1.7300400e-01, 6.7394490e+00],
[ 8.5276871e+00, 6.0145002e-02, 3.4448421e+00],
[ 4.2400112e+00, 1.1435485e+01, 5.5798002e-02]],
[[ 9.4334087e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 9.4334087e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 2.0947056e+01]],
[[-6.3495708e+00, 6.5125003e+00, 5.8539939e+00],
[ 6.0938725e+00, -6.4372239e+00, 5.4889545e+00],
[ 6.0836735e+00, 6.1772451e+00, -5.7762318e+00]],
[[ 6.8768296e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.1048177e+01, -2.1058772e+00],
[ 0.0000000e+00, -3.7062699e-03, 1.9015215e+01]],
[[ 4.9137821e+00, -8.5109196e+00, 0.0000000e+00],
[ 4.9137821e+00, 8.5109196e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 6.7629590e+00]],
[[ 6.3021402e+00, -6.3853199e-03, -1.0332731e+00],
[-1.7227671e+00, 6.0621023e+00, -1.0332731e+00],
[ 3.0169221e-02, 3.9895471e-02, 1.2250834e+01]],
[[ 7.4619780e+00, -1.4151999e-03, 4.3041716e+00],
[ 2.4855030e+00, 7.0358648e+00, 4.3041716e+00],
[-2.0012499e-03, -1.4151999e-03, 8.6143484e+00]],
[[ 7.9092207e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 7.5781007e+00, -1.6950918e+00],
[ 0.0000000e+00, -3.5751951e-01, 6.9183011e+00]],
[[ 6.7416244e+00, 5.3566109e-02, 4.4414983e+00],
[-1.2468886e+00, 7.0104933e+00, 1.9110441e+01],
[-1.6173536e-01, 2.7245510e-01, 1.0600656e+01]],
[[ 5.3063178e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 9.1126146e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0040712e+01]],
[[ 5.6279058e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 6.9537921e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0501416e+01]],
[[ 4.9857669e+00, 3.5377501e-03, -7.2275919e-01],
[ 1.6229881e+00, 5.5577154e+00, -1.0519388e+00],
[-4.2640892e-01, -7.8503363e-02, 7.8559442e+00]],
[[-4.6255998e-02, 4.7460151e+00, 3.0836339e+00],
[ 4.7340288e+00, -4.6193000e-02, -3.0838571e+00],
[-4.7798352e+00, 4.7923450e+00, -6.1672649e+00]],
[[ 6.1701546e+00, 1.2492373e-01, 4.7747571e-02],
[ 1.7339531e-01, 8.6193953e+00, -3.8590480e-02],
[ 4.4977799e-02, -2.7458400e-02, 5.7669253e+00]],
[[ 0.0000000e+00, 9.9999999e-09, -6.2029986e+00],
[-5.1461577e+00, 6.1716561e+00, -3.1015000e+00],
[ 5.1461577e+00, 6.1716561e+00, -3.1015000e+00]],
[[-3.9119451e+00, -3.1703908e+00, 2.1988914e+00],
[ 3.9119451e+00, -3.1703908e+00, -2.1988914e+00],
[ 3.8619483e+00, -3.1703908e+00, 5.5061889e+00]],
[[ 0.0000000e+00, -3.7546806e+00, 0.0000000e+00],
[-4.8117371e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -5.4126735e+00]],
[[ 2.7154329e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.2441120e+00, -3.8661101e-01],
[ 0.0000000e+00, 0.0000000e+00, 9.3888063e+00]],
[[-4.2616282e+00, -8.0763102e-03, -4.2752743e+00],
[-8.2218703e-03, -4.2585020e+00, -4.2751322e+00],
[-3.6448999e-03, -4.2542191e+00, 4.2708263e+00]],
[[ 5.3959742e+00, 2.2922444e-01, -4.5296320e-01],
[-7.0430361e-02, -1.8796436e+00, -6.2123251e+00],
[ 1.2653049e+00, 6.4092894e+00, 1.2955921e+00]],
[[ 4.1179528e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.8495660e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 5.8495660e+00]],
[[ 4.2995591e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 4.2995591e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0174184e+01]],
[[ 0.0000000e+00, 0.0000000e+00, -4.2354169e+00],
[-3.7717474e+00, -6.5328579e+00, 0.0000000e+00],
[-3.7717595e+00, 6.5328650e+00, 0.0000000e+00]],
[[-2.6921539e+00, 2.6921539e+00, 5.3483038e+00],
[ 2.6921539e+00, -2.6921539e+00, 5.3483038e+00],
[ 2.6921539e+00, 2.6921539e+00, -5.3483038e+00]],
[[-4.3139734e+00, 0.0000000e+00, 0.0000000e+00],
[-2.1569867e+00, -3.5078707e+00, 3.7477229e+00],
[ 2.1569867e+00, 3.4406567e+00, 3.6862457e+00]],
[[-2.1319211e+00, 2.1319211e+00, 5.1160870e+00],
[ 2.1319211e+00, -2.1319211e+00, 5.1160870e+00],
[ 2.1319211e+00, 2.1319211e+00, -5.1160870e+00]],
[[ 0.0000000e+00, 3.1339090e+00, 3.1339090e+00],
[ 3.1339090e+00, 0.0000000e+00, 3.1339090e+00],
[ 3.1339090e+00, 3.1339090e+00, 0.0000000e+00]],
[[ 1.8203124e+00, -3.1528738e+00, 0.0000000e+00],
[ 1.8203124e+00, 3.1528738e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 6.0473022e+00]],
[[ 0.0000000e+00, 2.5252669e+00, 2.5252669e+00],
[ 2.5252669e+00, 0.0000000e+00, 2.5252669e+00],
[ 2.5252669e+00, 2.5252669e+00, 0.0000000e+00]],
[[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 0.0000000e+00]]], dtype=float32):Rendering array...
,), target_data=e_form=Array([-6.45586 , -6.8128195, -6.207207 , -5.4671555, -6.7837076,
-5.1435127, -7.477968 , -8.923818 , -8.200472 , -7.9467635,
-9.577978 , -6.3009315, -6.341573 , -6.54503 , -9.78936 ,
-8.184659 , -6.8508086, -5.8719406, -4.560977 , -6.476559 ,
-8.410168 , -7.7028112, -4.6160893, -7.076347 , -4.2489 ,
-4.8315697, -4.6079087, -5.067987 , -4.573908 , -4.5372405,
-3.712647 , 0. ], dtype=float32):Rendering array...
, force=Array([[ 0.01931927, -0.00101004, 0.00514533],
[ 0.01053436, 0.01622596, 0.00514533],
[ 0.00878491, -0.017236 , 0.00514533],
...,
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ],
[ 0. , 0. , 0. ]], dtype=float32):Rendering array...
, stress=Array([[[-5.7650656e-01, 1.3470000e-05, 0.0000000e+00],
[-1.3470000e-05, -5.7650650e-01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -1.8034226e-01]],
[[ 1.2198363e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.0798175e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 7.2570233e+00]],
[[-6.6261089e-01, -1.3693068e+00, 1.7417374e-01],
[-1.3692706e+00, 9.3778890e-01, 5.2438201e-03],
[ 1.7419136e-01, 5.2425899e-03, -9.8206145e-01]],
[[-2.8648193e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -2.8648193e+01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -1.8271584e+01]],
[[-3.5376236e+00, -6.9719963e-02, -1.5628552e-01],
[-6.9719352e-02, -3.4879370e+00, -6.0669428e-01],
[-1.5628555e-01, -6.0669398e-01, -2.6510825e+00]],
[[ 4.2228210e-01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 3.6939880e-01, 2.1828409e-01],
[ 0.0000000e+00, 2.1827926e-01, 7.3054400e-03]],
[[ 5.8439370e-02, 2.6000001e-07, -0.0000000e+00],
[-2.6000001e-07, 5.8439370e-02, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -1.5984979e+00]],
[[-5.6706039e+01, 2.8917530e+00, -1.5447336e+01],
[ 2.8917530e+00, -5.5068783e+01, -2.0427404e+01],
[-1.5447323e+01, -2.0427385e+01, -2.5514915e+01]],
[[-5.2726287e-01, -5.5939171e-02, -1.3698542e-01],
[-5.5939171e-02, -4.8771665e-01, -9.6870460e-02],
[-1.3698542e-01, -9.6870460e-02, -6.8537790e-01]],
[[ 5.0960288e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 2.7661177e+01, 8.4236950e-01],
[ 0.0000000e+00, 8.4236366e-01, -8.4243935e-01]],
[[-3.1354368e+00, -2.1732990e-02, -1.6962990e-02],
[-2.1732951e-02, -3.8033817e+00, 3.7452817e+00],
[-1.6962999e-02, 3.7452817e+00, -1.9895161e+00]],
[[ 1.7607585e-01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.0304039e-01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.6413042e-01]],
[[-2.0985847e+00, -1.6143000e-04, -4.7509999e-05],
[-1.6518999e-04, -1.3785554e+00, -1.3842000e-04],
[-5.1700001e-05, -1.3332001e-04, 8.3491117e-01]],
[[-9.3856964e+00, -3.4404862e-01, -2.0616198e+00],
[-3.4404868e-01, -7.5551205e+00, -7.3723370e-01],
[-2.0616195e+00, -7.3723364e-01, -1.1444527e+01]],
[[-1.9861711e+00, -2.8908322e+00, 7.8711361e-01],
[-2.8947139e+00, -4.8037186e+00, 2.2988582e-01],
[ 7.8879166e-01, 2.2843994e-01, 2.1315180e+01]],
[[-1.4272230e+00, -1.6913697e+00, -5.8565631e+00],
[-1.6913695e+00, -4.9773030e+00, -1.3547415e+00],
[-5.8565636e+00, -1.3547415e+00, 1.7153270e+00]],
[[-6.0113930e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -8.9604294e+01, 7.7699997e-06],
[ 0.0000000e+00, 7.7699997e-06, -6.6187057e+01]],
[[ 2.7452025e+00, -0.0000000e+00, 4.3098168e+00],
[-0.0000000e+00, 9.5339581e-02, 0.0000000e+00],
[ 4.3098168e+00, 0.0000000e+00, -4.1897740e+00]],
[[-1.9521265e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -6.4751107e-01, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -7.5113559e-01]],
[[ 2.3531018e+02, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 2.1068205e+01, -1.0828968e+02],
[ 0.0000000e+00, -1.0828968e+02, 1.9093062e+02]],
[[-4.9796581e+00, -1.2185392e+00, -4.9542028e-01],
[-1.2196398e+00, -2.3808525e+00, -6.0947609e-01],
[-4.9716067e-01, -6.1348146e-01, -7.8811878e-01]],
[[ 8.0838509e+00, 1.0610949e+01, -6.8462425e-01],
[ 1.0610948e+01, 1.6280436e+01, -5.4885068e+00],
[-6.8462473e-01, -5.4885073e+00, -1.0486948e+01]],
[[ 1.4801001e+01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.6347537e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 5.6347537e+00]],
[[ 4.2995720e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 4.2995720e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 6.7877603e+00]],
[[-5.6527023e+00, -1.9787040e-02, 0.0000000e+00],
[-1.9787040e-02, -5.6755505e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, -7.4912200e+00]],
[[-1.5253462e+00, 0.0000000e+00, -0.0000000e+00],
[ 0.0000000e+00, -1.5253462e+00, -0.0000000e+00],
[-0.0000000e+00, -0.0000000e+00, 3.4477546e+00]],
[[-1.2089447e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 5.6878867e+00, -2.0085368e+00],
[ 0.0000000e+00, -2.0085406e+00, 3.9861345e+00]],
[[ 1.3792354e-01, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.3792354e-01, -0.0000000e+00],
[ 0.0000000e+00, -0.0000000e+00, 2.5862169e-01]],
[[ 1.4818479e+01, 1.4620157e+00, 1.4620157e+00],
[ 1.4620157e+00, 1.4818479e+01, 1.4620157e+00],
[ 1.4620157e+00, 1.4620157e+00, 1.4818479e+01]],
[[-3.0764341e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, -3.0764344e+00, 0.0000000e+00],
[-0.0000000e+00, 0.0000000e+00, 4.8425870e+00]],
[[-1.4238837e+00, 0.0000000e+00, -0.0000000e+00],
[ 0.0000000e+00, -1.4238837e+00, 0.0000000e+00],
[-0.0000000e+00, 0.0000000e+00, -1.4238837e+00]],
[[ 1.0000000e+00, 0.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 1.0000000e+00, 0.0000000e+00],
[ 0.0000000e+00, 0.0000000e+00, 1.0000000e+00]]], dtype=float32):Rendering array...
,),), 'energies': Array([-1.3220434e+00, -1.3318594e+00, -1.3613300e+00, ...,
-5.4650640e-44, -5.6051939e-44, -5.0446745e-44], dtype=float32):Rendering array...
,}children: 'reduction': reductioninput: 'data': Array([-8.599683, -8.602829, -8.612274, ..., 0. , 0. ,
0. ], dtype=float32):Rendering array...
, 'segments': Array([ 0, 0, 0, ..., 31, 31, 31], dtype=int32):Rendering array...
, 'num_segments': 32,}output: Array([-6.459625 , -6.8161387, -6.2099233, -5.4640856, -6.7733483,
-5.148916 , -7.4866095, -8.940761 , -8.207953 , -7.966073 ,
-9.58256 , -6.279612 , -6.3337655, -6.549827 , -9.774852 ,
-8.180584 , -6.8625493, -5.86331 , -4.5619845, -6.476076 ,
-8.405456 , -7.6989017, -4.6202683, -7.042893 , -4.2511187,
-4.8485074, -4.604096 , -5.0758014, -4.5818453, -4.4318686,
-3.673324 , 0. ], dtype=float32):Rendering array...
,),}params: 180 parametersscale: Array([4.2038954e-45, 3.0043125e-01, 4.1919053e-01, 3.2384700e-01,
3.2050502e-01, 3.0024481e-01, 2.6050037e-01, 4.1325513e-01,
2.9645354e-01, 2.7672523e-01, 7.3047429e-01, 2.7780551e-01,
4.0432665e-01, 2.6659670e-01, 2.9438594e-01, 4.3402749e-01,
3.4227425e-01, 2.9799023e-01, 7.4895459e-01, 2.5004971e-01,
3.5872504e-01, 2.8236023e-01, 6.4121890e-01, 3.9409107e-01,
7.4890226e-01, 3.5291594e-01, 3.1090504e-01, 3.4960249e-01,
4.6771184e-01, 5.9428304e-01, 2.5766170e-01, 2.8141865e-01,
4.9121195e-01, 3.2623026e-01, 6.4953142e-01, 6.3061160e-01,
8.5306853e-01, 6.3829052e-01, 7.6778311e-01, 2.8485122e-01,
2.8993279e-01, 2.7547282e-01, 9.6464241e-01, 7.5048459e-01,
8.1315958e-01, 7.9247254e-01, 4.5107767e-01, 4.5357430e-01,
4.8774320e-01, 3.5615650e-01, 2.5400338e-01, 3.0025351e-01,
3.7530035e-01, 6.5497804e-01, 2.7789482e-01, 3.1669718e-01,
4.0909296e-01, 3.8105384e-01, 7.5613052e-01, 7.5295144e-01,
8.0258620e-01, 7.9740274e-01, 1.5784103e-01, 6.9885850e-01,
7.6217997e-01, 5.1847076e-01, 3.0885923e-01, 5.0938869e-01,
5.6917423e-01, 7.5945461e-01, 6.3645500e-01, 5.9862715e-01,
5.5500871e-01, 7.4034333e-01, 3.0939424e-01, 5.0928408e-01,
7.7029741e-01, 2.4929152e-01, 7.7238590e-01, 7.6192188e-01,
6.0438299e-01, 1.8188974e+00, 1.0024991e+00, 7.7957290e-01,
7.7421695e-01, 1.9463260e+00, 8.6763585e-01, 6.9731098e-01,
8.5510230e-01, 4.2600378e-01], dtype=float32):Rendering array...
, shift: Array([ 0. , -8.254613 , -3.8793902 , -2.9042175 ,
-8.1759615 , -6.3587766 , -4.0522203 , -1.9677219 ,
-7.945281 , -3.5150082 , -8.059753 , -3.1420217 ,
-1.8107116 , -3.5583389 , -8.935444 , -9.42458 ,
-9.403597 , -5.4425173 , -6.7757645 , -4.5880127 ,
-5.1307225 , -4.574444 , -3.315387 , -1.9589221 ,
-6.82099 , -5.8368945 , -5.44436 , -8.114189 ,
-6.29539 , -3.7478168 , -1.1364459 , -3.721388 ,
-10.180324 , -6.538427 , -3.8525372 , -3.5311642 ,
-11.642563 , -10.856619 , -8.515128 , -2.2322743 ,
-0.6706738 , -3.884738 , -9.75878 , -10.200178 ,
-12.651368 , -7.912676 , -7.1350646 , -1.7933267 ,
-4.5861883 , -2.4339597 , -4.706932 , -3.4705129 ,
-7.2555256 , -9.282878 , -2.6069057 , -2.035008 ,
-3.1942008 , -6.1275 , -6.5005355 , -12.746008 ,
-7.2460003 , -13.921683 , -3.484169 , -15.289538 ,
-6.6684675 , -8.383291 , -2.104009 , -4.239751 ,
-11.927877 , -16.182364 , -12.089612 , -11.375482 ,
-9.446338 , -6.8297224 , -2.9690852 , -8.491434 ,
-6.6075172 , 0.03989757, -6.690451 , -6.6346316 ,
-9.987788 , 5.55694 , -11.5706415 , -6.5398293 ,
-6.751645 , 6.2271156 , -6.290777 , -0.219617 ,
-1.8358347 , -0.51835036], dtype=float32):Rendering array...
,}output: Array([[-6.459625 ],
[-6.8161387],
[-6.2099233],
[-5.4640856],
[-6.7733483],
[-5.148916 ],
[-7.4866095],
[-8.940761 ],
[-8.207953 ],
[-7.966073 ],
[-9.58256 ],
[-6.279612 ],
[-6.3337655],
[-6.549827 ],
[-9.774852 ],
[-8.180584 ],
[-6.8625493],
[-5.86331 ],
[-4.5619845],
[-6.476076 ],
[-8.405456 ],
[-7.6989017],
[-4.6202683],
[-7.042893 ],
[-4.2511187],
[-4.8485074],
[-4.604096 ],
[-5.0758014],
[-4.5818453],
[-4.4318686],
[-3.673324 ],
[ 0. ]], dtype=float32):Rendering array...
,),}output: Array([[-6.459625 ],
[-6.8161387],
[-6.2099233],
[-5.4640856],
[-6.7733483],
[-5.148916 ],
[-7.4866095],
[-8.940761 ],
[-8.207953 ],
[-7.966073 ],
[-9.58256 ],
[-6.279612 ],
[-6.3337655],
[-6.549827 ],
[-9.774852 ],
[-8.180584 ],
[-6.8625493],
[-5.86331 ],
[-4.5619845],
[-6.476076 ],
[-8.405456 ],
[-7.6989017],
[-4.6202683],
[-7.042893 ],
[-4.2511187],
[-4.8485074],
[-4.604096 ],
[-5.0758014],
[-4.5818453],
[-4.4318686],
[-3.673324 ],
[ 0. ]], dtype=float32):Rendering array...
,)